Home
last modified time | relevance | path

Searched refs:HandleReshape (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h213 Status HandleReshape(HloInstruction* reshape) override;
Dhlo_cost_analysis.h115 Status HandleReshape(const HloInstruction* reshape) override;
Ddfs_hlo_visitor_with_default.h204 Status HandleReshape(HloInstructionPtr reshape) override { in HandleReshape() function
Dhlo_verifier.h80 Status HandleReshape(HloInstruction* reshape) override;
Ddfs_hlo_visitor.h247 virtual Status HandleReshape(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc103 Status HandleReshape(HloInstruction* hlo) override;
816 Status DynamicDimensionInferenceVisitor::HandleReshape(HloInstruction* hlo) { in HandleReshape() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc512 Status HloCostAnalysis::HandleReshape(const HloInstruction*) { in HandleReshape() function in xla::HloCostAnalysis
Dhlo_verifier.cc729 Status ShapeVerifier::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::ShapeVerifier
Dhlo_evaluator.cc473 Status HloEvaluator::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::HloEvaluator
Dalgebraic_simplifier.cc341 Status HandleReshape(HloInstruction* reshape) override;
3872 Status AlgebraicSimplifierVisitor::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction.cc3203 return visitor->HandleReshape(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h437 Status HandleReshape(HloInstruction* hlo) override;
Dspmd_partitioner.cc1969 Status SpmdPartitioningVisitor::HandleReshape(HloInstruction* hlo) { in HandleReshape() function in xla::spmd::SpmdPartitioningVisitor