Searched refs:distributed_model (Results 1 – 1 of 1) sorted by relevance
768 distributed_model = strategy.extended.call_for_each_replica(771 set_distributed_model(model, mode, distributed_model)819 distributed_model = strategy.extended.call_for_each_replica(821 set_distributed_model(model, mode, distributed_model)823 model._make_callback_model(distributed_model)906 distributed_model = get_distributed_model(model, mode)915 if (distributed_model and hasattr(distributed_model, '_distribution_function')916 and not (hasattr(distributed_model, '_recompile_exec_function') and917 distributed_model._recompile_exec_function)):918 return distributed_model._distributed_function[all …]