Searched defs:InfeedWithToken (Results 1 – 1 of 1) sorted by relevance
1356 XlaOp XlaBuilder::InfeedWithToken(const XlaOp& token, const Shape& shape, in InfeedWithToken() function in xla::XlaBuilder3468 XlaOp InfeedWithToken(const XlaOp& token, const Shape& shape, in InfeedWithToken() function