Searched defs:global_batch_size (Results 1 – 5 of 5) sorted by relevance
348 def batch_sizes_for_worker(global_batch_size, num_workers, argument
148 def get_batch_size(global_batch_size, distribution): argument
42 def _test(self, global_batch_size, num_workers, num_replicas_per_worker, argument
413 global_batch_size=None): argument
517 def get_per_replica_batch_size(self, global_batch_size): argument