Searched refs:OutfeedWithToken (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 31 OutfeedWithToken(GetTupleElement(infeed, 0) + in BuildComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | outfeed_receiver.cc | 448 token = OutfeedWithToken( in AddOutfeedToBuilder()
451 token = OutfeedWithToken(data, token, shape_with_layout, ""); in AddOutfeedToBuilder()
|
| D | ops.cc | 180 ops.def("OutfeedWithToken", &OutfeedWithToken, py::arg("operand"), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 1818 ops.OutfeedWithToken(x, token, outfeed_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 97 auto outfeed = OutfeedWithToken(data, token, outfeed_shape, ""); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 421 // OutfeedOp corresponds to 'OutfeedWithToken' xla client API and not 'Outfeed'.
422 // OutfeedWithToken allows ordering of outfeed HLO instructions using tokens.
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 643 XlaOp OutfeedWithToken(XlaOp operand, XlaOp token,
1422 friend XlaOp OutfeedWithToken(XlaOp operand, XlaOp token,
2008 XlaOp OutfeedWithToken(XlaOp operand, XlaOp token,
|
| D | xla_builder.cc | 1820 XlaOp XlaBuilder::OutfeedWithToken(XlaOp operand, XlaOp token, in OutfeedWithToken() function in xla::XlaBuilder
4926 XlaOp OutfeedWithToken(const XlaOp operand, const XlaOp token, in OutfeedWithToken() function
4929 return operand.builder()->OutfeedWithToken(operand, token, shape_with_layout, in OutfeedWithToken()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 810 value_map[op] = xla::OutfeedWithToken( in ExportXlaOp()
|