Searched refs:RecvFromHost (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 242 const auto result_token_tuple = xla::RecvFromHost( in Compile()
511 xla::XlaOp result = xla::RecvFromHost(token, xla_shape, channel); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 300 ops.def("RecvFromHost", &RecvFromHost, py::arg("token"), py::arg("shape"), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 2711 data = ops.RecvFromHost(
2768 data = ops.RecvFromHost(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 905 XlaOp RecvFromHost(XlaOp token, const Shape& shape,
1501 friend XlaOp RecvFromHost(XlaOp token, const Shape& shape,
2646 XlaOp RecvFromHost(XlaOp token, const Shape& shape,
|
| D | xla_builder.cc | 3517 XlaOp XlaBuilder::RecvFromHost(XlaOp token, const Shape& shape, in RecvFromHost() function in xla::XlaBuilder
4950 XlaOp RecvFromHost(const XlaOp token, const Shape& shape, in RecvFromHost() function
4952 return token.builder()->RecvFromHost(token, shape, handle); in RecvFromHost()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1298 xla_result = xla::RecvFromHost(token, data_shape, in ExportXlaOp()
|