Searched refs:RecvFromHost (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 226 const auto result_token_tuple = xla::RecvFromHost( in Compile()
481 xla::XlaOp result = xla::RecvFromHost(token, xla_shape, channel); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 873 XlaOp RecvFromHost(XlaOp token, const Shape& shape,
1418 friend XlaOp RecvFromHost(XlaOp token, const Shape& shape,
2488 XlaOp RecvFromHost(XlaOp token, const Shape& shape,
|
| D | xla_builder.cc | 3285 XlaOp XlaBuilder::RecvFromHost(XlaOp token, const Shape& shape, in RecvFromHost() function in xla::XlaBuilder
4916 XlaOp RecvFromHost(const XlaOp token, const Shape& shape, in RecvFromHost() function
4918 return token.builder()->RecvFromHost(token, shape, handle); in RecvFromHost()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 844 value_map[op] = xla::RecvFromHost(token, xla::TypeToShape(result_type), in ExportXlaOp()
|