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