Searched defs:AllReduce (Results 1 – 3 of 3) sorted by relevance
71 ## AllReduce section in Operation Semantics
1566 TEST_P(ParameterizedMetadataTest, AllReduce) { in TEST_P() argument
2781 XlaOp XlaBuilder::AllReduce(XlaOp operand, const XlaComputation& computation, in AllReduce() function in xla::XlaBuilder4669 XlaOp AllReduce(const XlaOp operand, const XlaComputation& computation, in AllReduce() function