Home
last modified time | relevance | path

Searched refs:InfeedWithToken (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc29 XlaOp infeed = InfeedWithToken(CreateToken(&b), scalar_s32); in BuildComputation()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h423 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,
Dxla_builder.cc1356 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()