Searched defs:AllGather (Results 1 – 2 of 2) sorted by relevance
32 ## AllGather section in Operation Semantics
2714 XlaOp XlaBuilder::AllGather(XlaOp operand, int64 all_gather_dimension, in AllGather() function in xla::XlaBuilder4653 XlaOp AllGather(const XlaOp operand, int64 all_gather_dimension, in AllGather() function