Searched refs:RecvWithToken (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 881 XlaOp RecvWithToken(XlaOp token, const Shape& shape,
1413 friend XlaOp RecvWithToken(XlaOp token, const Shape& shape,
2474 XlaOp RecvWithToken(XlaOp token, const Shape& shape,
|
| D | xla_builder.cc | 3196 XlaOp recv = RecvWithToken(token, shape, handle); in Recv()
3210 XlaOp XlaBuilder::RecvWithToken(XlaOp token, const Shape& shape, in RecvWithToken() function in xla::XlaBuilder
4905 XlaOp RecvWithToken(const XlaOp token, const Shape& shape, in RecvWithToken() function
4907 return token.builder()->RecvWithToken(token, shape, handle); in RecvWithToken()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 848 value_map[op] = xla::RecvWithToken(token, xla::TypeToShape(result_type), in ExportXlaOp()
|