Home
last modified time | relevance | path

Searched defs:HandleReshape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h204 Status HandleReshape(HloInstructionPtr reshape) override { in HandleReshape() function
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
Ddynamic_dimension_inference.cc816 Status DynamicDimensionInferenceVisitor::HandleReshape(HloInstruction* hlo) { in HandleReshape() function in xla::DynamicDimensionInferenceVisitor
Dhlo_evaluator.cc473 Status HloEvaluator::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::HloEvaluator
Dalgebraic_simplifier.cc3872 Status AlgebraicSimplifierVisitor::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1969 Status SpmdPartitioningVisitor::HandleReshape(HloInstruction* hlo) { in HandleReshape() function in xla::spmd::SpmdPartitioningVisitor