Searched refs:get_distributed_model (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils.py | 920 distributed_model = get_distributed_model(model, mode)
936 distributed_model = get_distributed_model(model, mode)
965 _per_replica_function, args=(get_distributed_model(model, mode),))
1012 _per_replica_function, args=(get_distributed_model(model, mode),))
1036 distributed_model = get_distributed_model(original_model, mode)
1048 distributed_model = get_distributed_model(model, mode)
1070 distributed_model = get_distributed_model(model, mode)
1076 def get_distributed_model(model, mode): function
1179 distributed_model = get_distributed_model(model, mode)
1182 distributed_model = get_distributed_model(model, mode)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed.py | 93 args=(dist_utils.get_distributed_model(model, mode), mode))
336 dist_utils.get_distributed_model(model, mode), mode)
482 dist_utils.get_distributed_model(model, mode), mode)
|