Searched refs:SendToHost (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 166 xla::SendToHost(input_handles[i], token, xla_shape, channel)); in Compile()
430 xla::SendToHost(operand, token, xla_shape, channel); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 870 XlaOp SendToHost(XlaOp operand, XlaOp token, const Shape& shape_with_layout,
1415 friend XlaOp SendToHost(XlaOp operand, XlaOp token,
2482 XlaOp SendToHost(XlaOp operand, XlaOp token, const Shape& shape_with_layout,
|
| D | xla_builder.cc | 3238 XlaOp XlaBuilder::SendToHost(XlaOp operand, XlaOp token, in SendToHost() function in xla::XlaBuilder
4910 XlaOp SendToHost(const XlaOp operand, const XlaOp token, in SendToHost() function
4912 return operand.builder()->SendToHost(operand, token, shape_with_layout, in SendToHost()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 990 value_map[op] = xla::SendToHost(operand, token, in ExportXlaOp()
|