def single_core_graph(is_training, features, mems):
    model_fn = get_model_fn()
    model_ret = model_fn(features=features, labels=None, mems=mems, is_training=is_training)
    return model_ret
def single_core_graph(is_training, features, mems):
    model_fn = get_model_fn()
    model_ret = model_fn(features=features, labels=None, mems=mems, is_training=is_training)
    return model_ret