Searched refs:HandleWhile (Results 1 – 15 of 15) sorted by relevance
74 Status HandleWhile(HloInstruction* hlo) override;414 Status DynamicDimensionInferenceVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::DynamicDimensionInferenceVisitor
110 Status HandleWhile(const HloInstruction* xla_while) override;
188 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile() function
219 Status HandleWhile(HloInstruction* while_hlo) override;
86 Status HandleWhile(HloInstruction* xla_while) override;
244 virtual Status HandleWhile(HloInstructionPtr hlo) = 0;
646 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::ShapeVerifier1353 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::__anond588ae4e0811::InstructionVerifier
663 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile() function in xla::HloCostAnalysis
1319 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile() function in xla::HloEvaluator
2632 return visitor->HandleWhile(this); in Visit()
172 Status HandleWhile(HloInstruction* xla_while) override;
977 Status IrEmitterUnnested::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::gpu::IrEmitterUnnested
178 Status HandleWhile(HloInstruction* xla_while) override;
226 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::cpu::__anonaf93c4290211::CollectProfileCandidates
2296 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::cpu::IrEmitter