Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_training_utils.py28 def global_batch_size_supported(distribution_strategy): function
Dkeras_correctness_test_base.py157 not distributed_training_utils.global_batch_size_supported(distribution))
Ddistribute_strategy_test.py375 if not distributed_training_utils.global_batch_size_supported(distribution):
397 if not distributed_training_utils.global_batch_size_supported(distribution):
440 if not distributed_training_utils.global_batch_size_supported(distribution):
1674 if not distributed_training_utils.global_batch_size_supported(distribution):
Ddistributed_training_utils_v1.py488 use_per_replica_batch = not dist_utils.global_batch_size_supported(
/external/tensorflow/tensorflow/python/keras/engine/
Dinput_layer.py120 distributed_training_utils.global_batch_size_supported(strategy):
Dtraining_v1.py1730 distributed_training_utils.global_batch_size_supported(