Searched refs:distributed_function (Results 1 – 1 of 1) sorted by relevance
821 distributed_function = get_distributed_function(model, mode)822 if distributed_function:823 return distributed_function837 def distributed_function(input_fn): function851 distributed_function = def_function.function(distributed_function)854 return [out.numpy() for out in distributed_function(input_fn)]856 execution_function = distributed_function917 distributed_function = _make_eager_execution_function(model, mode)919 distributed_function = _make_graph_execution_function(model, mode)923 distributed_model._distributed_function = distributed_function[all …]