Searched defs:batch_reduce (Results 1 – 2 of 2) sorted by relevance
234 reduce_op, batch_reduce): argument397 batch_reduce): argument416 cross_device_ops_instance, batch_reduce): argument671 batch_reduce, argument786 def testReduceIndexedSlicesLocal(self, required_gpus, batch_reduce, argument
397 def batch_reduce(self, reduce_op, value_destination_pairs, options=None): member in CrossDeviceOps