Searched refs:distributed_model (Results 1 – 2 of 2) sorted by relevance
758 distributed_model = strategy.extended.call_for_each_replica(761 set_distributed_model(model, mode, distributed_model)809 distributed_model = strategy.extended.call_for_each_replica(811 set_distributed_model(model, mode, distributed_model)813 model._make_callback_model(distributed_model)896 distributed_model = get_distributed_model(model, mode)905 if (distributed_model and hasattr(distributed_model, '_distribution_function')906 and not (hasattr(distributed_model, '_recompile_exec_function') and907 distributed_model._recompile_exec_function)):908 return distributed_model._distributed_function[all …]
373 distributed_model = self.Model("value", file_path)374 func_captures = distributed_model.use_table.get_concrete_function(381 deferred_captures = distributed_model.use_table.get_concrete_function(385 self.verifyWorkerLocalInstance(coordinator, distributed_model)