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()
|