Searched defs:HandleWhile (Results 1 – 8 of 8) sorted by relevance
225 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile() function
1257 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::ShapeVerifier2167 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::__anonba98a3510d11::InstructionVerifier
989 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile() function in xla::HloCostAnalysis
1542 Status DynamicDimensionInferenceVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::DynamicDimensionInferenceVisitor
2095 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile() function in xla::HloEvaluator
281 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::cpu::__anond5cc82ac0311::CollectProfileCandidates
2380 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::cpu::IrEmitter
2827 Status SpmdPartitioningVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::spmd::SpmdPartitioningVisitor