Searched refs:InfeedWithToken (Results 1 – 8 of 8) 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/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 83 auto infeed = InfeedWithToken(token, ShapeUtil::MakeShape(F32, {4, 8}), ""); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 168 ops.def("InfeedWithToken", &InfeedWithToken, py::arg("token"), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 1773 ops.InfeedWithToken(
1792 ops.InfeedWithToken(
1810 x_and_token = ops.InfeedWithToken(
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 392 // InfeedOp corresponds to 'InfeedWithToken' xla client API and not 'Infeed'.
393 // InfeedWithToken allows ordering of infeed HLO instructions using tokens.
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 637 XlaOp InfeedWithToken(XlaOp token, const Shape& shape, const string& config);
1420 friend XlaOp InfeedWithToken(XlaOp token, const Shape& shape,
1993 XlaOp InfeedWithToken(XlaOp token, const Shape& shape,
|
| D | xla_builder.cc | 1736 XlaOp XlaBuilder::InfeedWithToken(XlaOp token, const Shape& shape, in InfeedWithToken() function in xla::XlaBuilder
4921 XlaOp InfeedWithToken(const XlaOp token, const Shape& shape, in InfeedWithToken() function
4923 return token.builder()->InfeedWithToken(token, shape, config); in InfeedWithToken()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 780 value_map[op] = xla::InfeedWithToken(token, xla::TypeToShape(result_type), in ExportXlaOp()
|