Searched defs:infeed (Results 1 – 14 of 14) sorted by relevance
113 XfeedQueueManager* infeed() { return &infeed_; } in infeed() function
326 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() local
58 const Literal* infeed = nullptr; member
259 Status HandleInfeed(HloInstruction* infeed) override { in HandleInfeed()
238 auto infeed = body_builder.AddInstruction( in TEST_F() local
115 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed()
247 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() local
792 auto infeed = false_builder.AddInstruction( in TEST_F() local
2162 auto infeed = in TEST_F() local
138 auto infeed = Infeed(&builder, infeed_shape); in TEST_F() local227 auto infeed = Infeed(&builder, infeed_shape); in TEST_F() local
29 XlaOp infeed = InfeedWithToken(CreateToken(&b), scalar_s32); in BuildComputation() local
125 for (const auto& infeed : xfeed_instrs) { in GetXfeedShape() local
1473 Status IrEmitterUnnested::HandleInfeed(HloInstruction* infeed) { in HandleInfeed()
1326 XlaOp infeed; in Infeed() local