Searched refs:get_per_replica_batch_size (Results 1 – 9 of 9) sorted by relevance
22 name: "get_per_replica_batch_size"
294 input_context.get_per_replica_batch_size(4))346 batch_size = ctx.get_per_replica_batch_size(8)407 batch_size = ctx.get_per_replica_batch_size(8)469 batch_size = ctx.get_per_replica_batch_size(8)
202 batch_size = input_context.get_per_replica_batch_size(global_batch_size)246 batch_size = input_context.get_per_replica_batch_size(global_batch_size)
571 self.assertEqual(2, input_context.get_per_replica_batch_size(12))573 input_context.get_per_replica_batch_size(13)
1046 batch_size = ctx.get_per_replica_batch_size(global_batch_size)1167 batch_size = ctx.get_per_replica_batch_size(global_batch_size)1215 batch_size = ctx.get_per_replica_batch_size(global_batch_size)
500 def get_per_replica_batch_size(self, global_batch_size): member in InputContext
66 batch_size = input_context.get_per_replica_batch_size(global_batch_size)
98 batch_size = input_context.get_per_replica_batch_size(global_batch_size)