Searched defs:outfeed (Results 1 – 12 of 12) sorted by relevance
114 XfeedQueueManager* outfeed() { return &outfeed_; } in outfeed() function
515 Status IrEmitter::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed()
95 Literal outfeed(ShapeUtil::MakeShape(S32, {})); in TestWithDeviceCount() local
286 Status HandleOutfeed(HloInstruction* outfeed) override { in HandleOutfeed()
137 Status HandleOutfeed(HloInstructionPtr outfeed) override { in HandleOutfeed()
816 HloInstruction* outfeed = in TEST_F() local
352 Status HloCostAnalysis::HandleOutfeed(const HloInstruction* outfeed) { in HandleOutfeed()
432 HloOutfeedInstruction* outfeed = Cast<HloOutfeedInstruction>(instruction); in HandleOutfeed() local
97 auto outfeed = OutfeedWithToken(data, token, outfeed_shape, ""); in TEST_F() local
285 Literal outfeed(*outfeed_shape); in ReplayComputation() local
1119 const HloOutfeedInstruction* outfeed = in EmitOutfeedOp() local
3209 Status IrEmitterUnnested::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed()