Searched refs:batch_reduce_implementation (Results 1 – 9 of 9) sorted by relevance
287 return self.batch_reduce_implementation(reduce_op, value_destination_pairs)326 def batch_reduce_implementation(self, reduce_op, value_destination_pairs): member in CrossDeviceOps395 def batch_reduce_implementation(self, reduce_op, value_destination_pairs): member in ReductionToOneDevice674 def batch_reduce_implementation(self, reduce_op, value_destination_pairs): member in AllReduceCrossDeviceOps997 def batch_reduce_implementation(self, reduce_op, value_destination_pairs): member in CollectiveAllReduce
14 name: "batch_reduce_implementation"
16 name: "batch_reduce_implementation"
15 name: "batch_reduce_implementation"