Searched refs:HandleWhile (Results 1 – 17 of 17) sorted by relevance
241 Status HandleWhile(HloInstruction* while_hlo) override;
120 Status HandleWhile(const HloInstruction* xla_while) override;
210 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile() function
95 Status HandleWhile(HloInstruction* xla_while) override;
261 virtual Status HandleWhile(HloInstructionPtr hlo) = 0;
139 Status HandleWhile(HloInstruction* hlo) override;1429 Status DynamicDimensionInferenceVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::DynamicDimensionInferenceVisitor
920 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::ShapeVerifier1815 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::__anon45533f2c0811::InstructionVerifier
960 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile() function in xla::HloCostAnalysis
2031 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile() function in xla::HloEvaluator
3231 return visitor->HandleWhile(this); in Visit()
164 Status HandleWhile(HloInstruction* xla_while) override;
262 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::cpu::__anon31d794070311::CollectProfileCandidates
2357 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::cpu::IrEmitter
425 Status HandleWhile(HloInstruction* hlo) override;
2837 Status SpmdPartitioningVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::spmd::SpmdPartitioningVisitor
195 Status HandleWhile(HloInstruction* xla_while) override;
2431 Status IrEmitterUnnested::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::gpu::IrEmitterUnnested