Searched refs:distributed_function (Results 1 – 1 of 1) sorted by relevance
831 distributed_function = get_distributed_function(model, mode)832 if distributed_function:833 return distributed_function847 def distributed_function(input_fn): function861 distributed_function = def_function.function(distributed_function)864 return [out.numpy() for out in distributed_function(input_fn)]866 execution_function = distributed_function927 distributed_function = _make_eager_execution_function(model, mode)929 distributed_function = _make_graph_execution_function(model, mode)933 distributed_model._distributed_function = distributed_function[all …]