Searched defs:HandleReshape (Results 1 – 10 of 10) sorted by relevance
28 Status HandleReshape(HloInstruction* reshape) override { in HandleReshape() function in xla::__anon20d8d4240111::ReshapeDecomposerVisitor
258 Status HandleReshape(HloInstruction* hlo) override { in HandleReshape() function in xla::__anon314bcc100111::LayoutNormalizationVisitor
217 Status HandleReshape(HloInstructionPtr reshape) override { in HandleReshape() function
1099 Status ShapeVerifier::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::ShapeVerifier2479 Status HandleReshape(HloInstruction* hlo) override { in HandleReshape() function in xla::__anonb51dd00a0e11::InstructionVerifier
522 Status HloCostAnalysis::HandleReshape(const HloInstruction*) { in HandleReshape() function in xla::HloCostAnalysis
925 Status DynamicDimensionInferenceVisitor::HandleReshape(HloInstruction* hlo) { in HandleReshape() function in xla::DynamicDimensionInferenceVisitor
1128 Status HloEvaluator::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::HloEvaluator
4248 Status AlgebraicSimplifierVisitor::HandleReshape(HloInstruction* reshape) { in HandleReshape() function in xla::AlgebraicSimplifierVisitor
274 static void HandleReshape(int width, int height) { in HandleReshape() function
2607 Status SpmdPartitioningVisitor::HandleReshape(HloInstruction* hlo) { in HandleReshape() function in xla::spmd::SpmdPartitioningVisitor