Searched refs:InfeedWithToken (Results 1 – 3 of 3) 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 | 423 XlaOp InfeedWithToken(const XlaOp& token, const Shape& shape,
998 friend XlaOp InfeedWithToken(const XlaOp& token, const Shape& shape,
1400 XlaOp InfeedWithToken(const XlaOp& token, const Shape& shape,
|
| D | xla_builder.cc | 1356 XlaOp XlaBuilder::InfeedWithToken(const XlaOp& token, const Shape& shape, in InfeedWithToken() function in xla::XlaBuilder
3468 XlaOp InfeedWithToken(const XlaOp& token, const Shape& shape, in InfeedWithToken() function
3470 return token.builder()->InfeedWithToken(token, shape, config); in InfeedWithToken()
|