Searched defs:AllToAllTuple (Results 1 – 1 of 1) sorted by relevance
3012 XlaOp XlaBuilder::AllToAllTuple(XlaOp operand, int64_t split_dimension, in AllToAllTuple() function in xla::XlaBuilder4547 XlaOp AllToAllTuple(const XlaOp operand, int64_t split_dimension, in AllToAllTuple() function