Searched defs:batch_reduce (Results 1 – 2 of 2) sorted by relevance
237 reduce_op, batch_reduce): argument400 batch_reduce): argument419 cross_device_ops_instance, batch_reduce): argument672 batch_reduce, argument782 def testReduceIndexedSlicesLocal(self, required_gpus, batch_reduce, argument
381 def batch_reduce(self, reduce_op, value_destination_pairs, options=None): member in CrossDeviceOps