Searched refs:distributed_model (Results 1 – 1 of 1) sorted by relevance
674 distributed_model = strategy.extended.call_for_each_replica(677 set_distributed_model(model, mode, distributed_model)725 distributed_model = strategy.extended.call_for_each_replica(727 set_distributed_model(model, mode, distributed_model)729 model._make_callback_model(distributed_model)736 distributed_model = get_distributed_model(model, mode)739 if distributed_model:740 return distributed_model._distributed_function750 distributed_model = get_distributed_model(model, mode)751 assert distributed_model[all …]