Searched defs:global_batch_size (Results 1 – 4 of 4) sorted by relevance
365 def batch_sizes_for_worker(global_batch_size, num_workers, argument
38 def _test(self, global_batch_size, num_workers, num_replicas_per_worker, argument
413 global_batch_size=None): argument
524 def get_per_replica_batch_size(self, global_batch_size): argument