Searched defs:HandleWhile (Results 1 – 8 of 8) sorted by relevance
223 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile() function
1290 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::ShapeVerifier2396 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::__anonb51dd00a0e11::InstructionVerifier
1018 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile() function in xla::HloCostAnalysis
1685 Status DynamicDimensionInferenceVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::DynamicDimensionInferenceVisitor
3300 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile() function in xla::HloEvaluator
396 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::cpu::__anonb41b63e50511::CollectProfileCandidates
2515 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::cpu::IrEmitter
3608 Status SpmdPartitioningVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::spmd::SpmdPartitioningVisitor