Home
last modified time | relevance | path

Searched defs:HandleWhile (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddfs_hlo_visitor_with_default.h223 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile() function
H A Dhlo_verifier.cc1290 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::ShapeVerifier
2396 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::__anon411d6f1b0e11::InstructionVerifier
H A Dhlo_cost_analysis.cc1018 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile() function in xla::HloCostAnalysis
H A Ddynamic_dimension_inference.cc1685 Status DynamicDimensionInferenceVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::DynamicDimensionInferenceVisitor
H A Dhlo_evaluator.cc3300 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile() function in xla::HloEvaluator
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.cc396 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile() function in xla::cpu::__anon660623760511::CollectProfileCandidates
H A Dir_emitter.cc2515 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile() function in xla::cpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3608 Status SpmdPartitioningVisitor::HandleWhile(HloInstruction* hlo) { in HandleWhile() function in xla::spmd::SpmdPartitioningVisitor