Home
last modified time | relevance | path

Searched refs:RecvWithToken (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h881 XlaOp RecvWithToken(XlaOp token, const Shape& shape,
1413 friend XlaOp RecvWithToken(XlaOp token, const Shape& shape,
2474 XlaOp RecvWithToken(XlaOp token, const Shape& shape,
Dxla_builder.cc3196 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/
Dmlir_hlo_to_hlo.cc848 value_map[op] = xla::RecvWithToken(token, xla::TypeToShape(result_type), in ExportXlaOp()