Searched refs:get_distributed_model (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | distributed_training_utils_v1.py | 896 distributed_model = get_distributed_model(model, mode)
912 distributed_model = get_distributed_model(model, mode)
941 _per_replica_function, args=(get_distributed_model(model, mode),))
988 _per_replica_function, args=(get_distributed_model(model, mode),))
1012 distributed_model = get_distributed_model(original_model, mode)
1024 distributed_model = get_distributed_model(model, mode)
1046 distributed_model = get_distributed_model(model, mode)
1052 def get_distributed_model(model, mode): function
1124 distributed_model = get_distributed_model(model, mode)
1127 distributed_model = get_distributed_model(model, mode)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training_distributed_v1.py | 88 args=(dist_utils.get_distributed_model(model, mode), mode))
331 dist_utils.get_distributed_model(model, mode), mode)
478 dist_utils.get_distributed_model(model, mode), mode)
|