Searched refs:HandleReshape (Results 1 – 11 of 11) sorted by relevance
52 Status HandleReshape(HloInstruction* hlo) override;330 Status DynamicDimensionInferenceVisitor::HandleReshape(HloInstruction* hlo) { in HandleReshape() function in xla::DynamicDimensionInferenceVisitor
106 Status HandleReshape(const HloInstruction* reshape) override;
182 Status HandleReshape(HloInstructionPtr reshape) override { in HandleReshape() function
197 Status HandleReshape(HloInstruction* reshape) override;
72 Status HandleReshape(HloInstruction* reshape) override;
231 virtual Status HandleReshape(HloInstructionPtr hlo) = 0;
392 Status HloCostAnalysis::HandleReshape(const HloInstruction*) { in HandleReshape() function in xla::HloCostAnalysis
494 Status ShapeVerifier::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::ShapeVerifier
436 Status HloEvaluator::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::HloEvaluator
226 Status HandleReshape(HloInstruction* reshape) override;2588 Status AlgebraicSimplifierVisitor::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::AlgebraicSimplifierVisitor
2610 return visitor->HandleReshape(this); in Visit()