Searched refs:xla_infeed (Results 1 – 2 of 2) sorted by relevance
196 Status HandleInfeed(HloInstruction* xla_infeed) override;
3190 Status IrEmitterUnnested::HandleInfeed(HloInstruction* xla_infeed) { in HandleInfeed() argument3191 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(xla_infeed)); in HandleInfeed()