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()
|