Searched refs:InfeedWithTokenInternal (Results 1 – 4 of 4) sorted by relevance
227 StatusOr<XlaOp> InfeedWithTokenInternal(const Shape& infeed_instruction_shape,
467 StatusOr<XlaOp> MlirHloBuilder::InfeedWithTokenInternal( in InfeedWithTokenInternal() function in xla::MlirHloBuilder
1756 return InfeedWithTokenInternal(infeed_instruction_shape, token, config); in InfeedWithToken()1760 StatusOr<XlaOp> XlaBuilder::InfeedWithTokenInternal( in InfeedWithTokenInternal() function in xla::XlaBuilder
638 virtual StatusOr<XlaOp> InfeedWithTokenInternal(