Searched defs:outfeed (Results 1 – 12 of 12) sorted by relevance
113 XfeedQueueManager* outfeed() { return &outfeed_; } in outfeed() function
529 Status IrEmitter::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed()
95 Literal outfeed(ShapeUtil::MakeShape(S32, {})); in TestWithDeviceCount() local
359 Status HandleOutfeed(HloInstruction* outfeed) override { in HandleOutfeed()
150 Status HandleOutfeed(HloInstructionPtr outfeed) override { in HandleOutfeed()
952 HloInstruction* outfeed = in TEST_F() local
352 Status HloCostAnalysis::HandleOutfeed(const HloInstruction* outfeed) { in HandleOutfeed()
782 HloOutfeedInstruction* outfeed = Cast<HloOutfeedInstruction>(instruction); in HandleOutfeed() local
2363 HloInstruction* outfeed = FindInstruction(module.get(), "outfeed"); in TEST_F() local
107 auto outfeed = OutfeedWithToken(data, token, outfeed_shape, ""); in TEST_F() local
326 Literal outfeed(*outfeed_shape); in ReplayComputation() local
1278 const HloOutfeedInstruction* outfeed = in EmitOutfeedOp() local