Home
last modified time | relevance | path

Searched refs:distributed_model (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Ddistributed_training_utils.py674 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_function
750 distributed_model = get_distributed_model(model, mode)
751 assert distributed_model
[all …]