Searched refs:AllToAllTuple (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 751 XlaOp AllToAllTuple(XlaOp operand, int64 split_dimension,
1311 friend XlaOp AllToAllTuple(XlaOp operand, int64 split_dimension,
2228 XlaOp AllToAllTuple(XlaOp operand, int64 split_dimension,
|
| D | xla_builder.cc | 2864 return AllToAllTuple(operand, split_dimension, concat_dimension, in AllToAll()
2923 XlaOp XlaBuilder::AllToAllTuple(XlaOp operand, int64 split_dimension, in AllToAllTuple() function in xla::XlaBuilder
4685 XlaOp AllToAllTuple(const XlaOp operand, int64 split_dimension, in AllToAllTuple() function
4689 return operand.builder()->AllToAllTuple(operand, split_dimension, in AllToAllTuple()
|