Home
last modified time | relevance | path

Searched refs:AllToAllTuple (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h751 XlaOp AllToAllTuple(XlaOp operand, int64 split_dimension,
1311 friend XlaOp AllToAllTuple(XlaOp operand, int64 split_dimension,
2228 XlaOp AllToAllTuple(XlaOp operand, int64 split_dimension,
Dxla_builder.cc2864 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()