Searched refs:InfeedWithToken (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 29 XlaOp infeed = InfeedWithToken(CreateToken(&b), scalar_s32); in BuildComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 469 XlaOp InfeedWithToken(XlaOp token, const Shape& shape,
1032 friend XlaOp InfeedWithToken(XlaOp token, const Shape& shape,
1488 XlaOp InfeedWithToken(XlaOp token, const Shape& shape,
|
| D | xla_builder.cc | 1413 XlaOp XlaBuilder::InfeedWithToken(XlaOp token, const Shape& shape, in InfeedWithToken() function in xla::XlaBuilder
3536 XlaOp InfeedWithToken(const XlaOp token, const Shape& shape, in InfeedWithToken() function
3538 return token.builder()->InfeedWithToken(token, shape, config); in InfeedWithToken()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | hlo_ops.td | 350 // InfeedOp corresponds to 'InfeedWithToken' xla client API and not 'Infeed'.
351 // InfeedWithToken allows ordering of infeed HLO instructions using tokens.
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 795 return ops.InfeedWithToken(token,
|
| D | xla.cc | 877 ops.def("InfeedWithToken", &InfeedWithToken, py::arg("token"), in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 565 xla::InfeedWithToken(value_map[op.token()], xla::TypeToShape(result_type), in ExportXlaOp()
|