Searched refs:distributed_model (Results 1 – 1 of 1) sorted by relevance
781 distributed_model = strategy.extended.call_for_each_replica(784 set_distributed_model(model, mode, distributed_model)832 distributed_model = strategy.extended.call_for_each_replica(834 set_distributed_model(model, mode, distributed_model)836 model._make_callback_model(distributed_model)920 distributed_model = get_distributed_model(model, mode)929 if (distributed_model and hasattr(distributed_model, '_distribution_function')930 and not (hasattr(distributed_model, '_recompile_exec_function') and931 distributed_model._recompile_exec_function)):932 return distributed_model._distributed_function[all …]