Searched defs:AllReduce (Results 1 – 5 of 5) sorted by relevance
71 ## AllReduce section in Operation Semantics
313 TEST_F(HloControlFlowFlatteningTest, AllReduce) { in TEST_F() argument
1593 struct AllReduce { struct1601 static AllReduce Handler() { return AllReduce(); } in Handler()1640 static bool AllReduce(runtime::KernelContext* ctx, void** args, void** attrs) { in AllReduce() function
1775 TEST_P(ParameterizedMetadataTest, AllReduce) { in TEST_P() argument
2936 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