microsoft/onnxscript

[Optimizer] Fold Shape -> {Slice, Gather} -> Concat -> Reshape

Ouverte

#2 736 ouverte le 17 déc. 2025

 (2 commentaires) (0 réaction) (0 personne assignée)Python (117 forks)auto 404
contribution welcomegood first issuemodule: rewriter

Métriques du dépôt

Stars
 (445 étoiles)
Métriques de merge PR
 (Merge moyen 4j 18h) (13 PRs mergées en 30 j)

Description

In Shape -> {Slice, Gather} -> Concat -> Reshape, when the dynamic axis coming from Shape is a single axis and the rest of the dimensions are not -1, we can replace the shape for Reshape to have a static -1 axis, and eliminate the nodes for dynamic shape evaluation.

Guide contributeur