Searched defs:AllToAllTuple (Results 1 – 2 of 2) sorted by relevance
3145 XlaOp XlaBuilder::AllToAllTuple(absl::Span<const XlaOp> operands, in AllToAllTuple() function in xla::XlaBuilder3183 XlaOp XlaBuilder::AllToAllTuple(XlaOp operand, int64_t split_dimension, in AllToAllTuple() function in xla::XlaBuilder4698 XlaOp AllToAllTuple(absl::Span<const XlaOp> operands, in AllToAllTuple() function4705 XlaOp AllToAllTuple(const XlaOp operand, int64_t split_dimension, in AllToAllTuple() function
467 TEST_F(XlaBuilderTest, AllToAllTuple) { in TEST_F() argument