Searched refs:global_batch_size_supported (Results 1 – 4 of 4) sorted by relevance
389 def global_batch_size_supported(distribution_strategy): function434 use_per_replica_batch = not global_batch_size_supported(
485 if not distributed_training_utils.global_batch_size_supported(distribution):517 if not distributed_training_utils.global_batch_size_supported(distribution):563 if not distributed_training_utils.global_batch_size_supported(distribution):1245 if not distributed_training_utils.global_batch_size_supported(distribution):
138 not distributed_training_utils.global_batch_size_supported(distribution))
221 not distributed_training_utils.global_batch_size_supported(940 if not distributed_training_utils.global_batch_size_supported(