Searched refs:RecvWithToken (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 566 XlaOp RecvWithToken(const XlaOp& token, const Shape& shape,
991 friend XlaOp RecvWithToken(const XlaOp& token, const Shape& shape,
1847 XlaOp RecvWithToken(const XlaOp& token, const Shape& shape,
|
| D | xla_builder.cc | 2364 XlaOp recv = RecvWithToken(token, shape, handle); in Recv()
2378 XlaOp XlaBuilder::RecvWithToken(const XlaOp& token, const Shape& shape, in RecvWithToken() function in xla::XlaBuilder
3452 XlaOp RecvWithToken(const XlaOp& token, const Shape& shape, in RecvWithToken() function
3454 return token.builder()->RecvWithToken(token, shape, handle); in RecvWithToken()
|