Searched refs:SendToHost (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 595 XlaOp SendToHost(XlaOp operand, XlaOp token, const Shape& shape_with_layout,
1027 friend XlaOp SendToHost(XlaOp operand, XlaOp token,
1923 XlaOp SendToHost(XlaOp operand, XlaOp token, const Shape& shape_with_layout,
|
| D | xla_builder.cc | 2436 XlaOp XlaBuilder::SendToHost(XlaOp operand, XlaOp token, in SendToHost() function in xla::XlaBuilder
3525 XlaOp SendToHost(const XlaOp operand, const XlaOp token, in SendToHost() function
3527 return operand.builder()->SendToHost(operand, token, shape_with_layout, in SendToHost()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 731 xla::SendToHost(value_map[op.operand()], value_map[op.token()], in ExportXlaOp()
|