Searched refs:dist_utils (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed_v1.py | 34 from tensorflow.python.keras.distribute import distributed_training_utils_v1 as dist_utils unknown
53 dist_utils.clone_model_on_replicas(
56 dist_utils._build_distributed_network(model, strategy, mode, inputs,
93 args=(dist_utils.get_distributed_model(model, mode), mode))
95 all_session_args) = dist_utils.unwrap_values(strategy, grouped_inputs,
175 iterator = dist_utils.get_iterator(dataset, current_strategy)
177 scope = dist_utils.distributed_scope(
202 dist_utils._copy_weights_to_distributed_model(model, mode)
228 dist_utils._reset_metrics(model)
266 dist_utils._copy_weights_to_original_model(model, ModeKeys.TRAIN)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 40 from tensorflow.python.keras.distribute import distributed_training_utils as dist_utils unknown
488 use_per_replica_batch = not dist_utils.global_batch_size_supported(
|