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