Searched refs:AllToAllArray (Results 1 – 2 of 2) sorted by relevance
2867 return AllToAllArray(operand, split_dimension, concat_dimension, split_count, in AllToAll()2871 XlaOp XlaBuilder::AllToAllArray( in AllToAllArray() function in xla::XlaBuilder
1443 XlaOp AllToAllArray(XlaOp operand, int64 split_dimension,