Searched defs:global_batch_size (Results 1 – 3 of 3) sorted by relevance
133 def get_batch_size(global_batch_size, distribution): argument
457 def global_batch_size(self): member in _InternalTPUContext
269 def get_per_replica_batch_size(self, global_batch_size): argument