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