Searched refs:global_batch_size_supported (Results 1 – 6 of 6) sorted by relevance
28 def global_batch_size_supported(distribution_strategy): function
157 not distributed_training_utils.global_batch_size_supported(distribution))
375 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):
488 use_per_replica_batch = not dist_utils.global_batch_size_supported(
120 distributed_training_utils.global_batch_size_supported(strategy):
1730 distributed_training_utils.global_batch_size_supported(