Searched refs:OutfeedWithToken (Results 1 – 3 of 3) 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/client/ |
| D | xla_builder.h | 428 XlaOp OutfeedWithToken(const XlaOp& operand, const XlaOp& token,
1000 friend XlaOp OutfeedWithToken(const XlaOp& operand, const XlaOp& token,
1415 XlaOp OutfeedWithToken(const XlaOp& operand, const XlaOp& token,
|
| D | xla_builder.cc | 1437 XlaOp XlaBuilder::OutfeedWithToken(const XlaOp& operand, const XlaOp& token, in OutfeedWithToken() function in xla::XlaBuilder
3473 XlaOp OutfeedWithToken(const XlaOp& operand, const XlaOp& token, in OutfeedWithToken() function
3476 return operand.builder()->OutfeedWithToken(operand, token, shape_with_layout, in OutfeedWithToken()
|