Searched defs:AllToAllTuple (Results 1 – 1 of 1) sorted by relevance
2923 XlaOp XlaBuilder::AllToAllTuple(XlaOp operand, int64 split_dimension, in AllToAllTuple() function in xla::XlaBuilder4685 XlaOp AllToAllTuple(const XlaOp operand, int64 split_dimension, in AllToAllTuple() function