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