Searched refs:get_per_replica_batch_size (Results 1 – 4 of 4) sorted by relevance
161 self.assertEqual(2, input_context.get_per_replica_batch_size(12))163 input_context.get_per_replica_batch_size(13)
269 def get_per_replica_batch_size(self, global_batch_size): member in InputContext
22 name: "get_per_replica_batch_size"