Searched refs:HandleInfeed (Results 1 – 16 of 16) sorted by relevance
84 Status HandleInfeed(HloInstruction* infeed) override;
196 Status HandleInfeed(HloInstruction* xla_infeed) override;
602 Status IrEmitter::HandleInfeed(HloInstruction*) { in HandleInfeed() function in xla::gpu::IrEmitter
3190 Status IrEmitterUnnested::HandleInfeed(HloInstruction* xla_infeed) { in HandleInfeed() function in xla::gpu::IrEmitterUnnested
283 Status HandleInfeed(HloInstruction* infeed) override { in HandleInfeed() function
87 Status HandleInfeed(const HloInstruction* infeed) override;
134 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed() function
68 Status HandleInfeed(HloInstruction*) override;
234 virtual Status HandleInfeed(HloInstructionPtr hlo) = 0;
339 Status HloCostAnalysis::HandleInfeed(const HloInstruction* infeed) { in HandleInfeed() function in xla::HloCostAnalysis
421 Status ShapeVerifier::HandleInfeed(HloInstruction* instruction) { in HandleInfeed() function in xla::ShapeVerifier
3221 return visitor->HandleInfeed(this); in Visit()
144 Status HandleInfeed(HloInstruction* infeed) override;
389 Status IrEmitter::HandleInfeed(HloInstruction* instruction) { in HandleInfeed() function in xla::cpu::IrEmitter
419 Status HandleInfeed(HloInstruction* hlo) override;
2491 Status SpmdPartitioningVisitor::HandleInfeed(HloInstruction* hlo) { in HandleInfeed() function in xla::spmd::SpmdPartitioningVisitor