Searched defs:AllReduce (Results 1 – 3 of 3) sorted by relevance
32 ## AllReduce section in Operation Semantics
1008 def AllReduce(self, operand, computation, replica_groups=None): member in ComputationBuilder
2113 XlaOp XlaBuilder::AllReduce(XlaOp operand, const XlaComputation& computation, in AllReduce() function in xla::XlaBuilder3306 XlaOp AllReduce(const XlaOp operand, const XlaComputation& computation, in AllReduce() function