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/engine/
Dinput_layer.py99 distributed_training_utils.global_batch_size_supported(strategy):
Dtraining.py1574 distributed_training_utils.global_batch_size_supported(
Dtraining_v1.py1727 distributed_training_utils.global_batch_size_supported(
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_training_utils.py435 def global_batch_size_supported(distribution_strategy): function
503 use_per_replica_batch = not global_batch_size_supported(
Ddistribute_strategy_test.py332 if not distributed_training_utils.global_batch_size_supported(distribution):
354 if not distributed_training_utils.global_batch_size_supported(distribution):
397 if not distributed_training_utils.global_batch_size_supported(distribution):
1533 if not distributed_training_utils.global_batch_size_supported(distribution):
Dkeras_correctness_test_base.py152 not distributed_training_utils.global_batch_size_supported(distribution))