Searched defs:AllGather (Results 1 – 5 of 5) sorted by relevance
32 ## AllGather section in Operation Semantics
372 TEST_F(HloControlFlowFlatteningTest, AllGather) { in TEST_F() argument
142 class AllGather: class
1858 struct AllGather { struct1866 static AllGather Handler() { return AllGather(); } in Handler()1902 static bool AllGather(runtime::KernelContext* ctx, void** args, void** attrs) { in AllGather() function
2868 XlaOp XlaBuilder::AllGather(XlaOp operand, int64_t all_gather_dimension, in AllGather() function in xla::XlaBuilder4653 XlaOp AllGather(const XlaOp operand, int64_t all_gather_dimension, in AllGather() function