Searched defs:AllReduce (Results 1 – 4 of 4) sorted by relevance
248 TEST_F(HloControlFlowFlatteningTest, AllReduce) { in TEST_F() argument
71 ## AllReduce section in Operation Semantics
1660 TEST_P(ParameterizedMetadataTest, AllReduce) { in TEST_P() argument
2810 XlaOp XlaBuilder::AllReduce(XlaOp operand, const XlaComputation& computation, in AllReduce() function in xla::XlaBuilder4520 XlaOp AllReduce(const XlaOp operand, const XlaComputation& computation, in AllReduce() function