You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I've been using ONNXScript extensively lately and really appreciate how powerful and flexible it is. As part of that experience, I and @Johansmm would like to suggest a few optimization patterns that could improve performance and simplify common model graphs. These are generally applicable across many models:
I've been using ONNXScript extensively lately and really appreciate how powerful and flexible it is. As part of that experience, I and @Johansmm would like to suggest a few optimization patterns that could improve performance and simplify common model graphs. These are generally applicable across many models:
BatchNormalizationinto preceding nodes (Conv,ConvTranspose,Gemm) Rewriter: Fold Batchnorm nodes #2312MatMul + AddintoGemm[Rewriter]: Add ∘ MatMul -> Gemm #2356Min + MaxintoClip[Rewriter]: add fusion rules for successive Min/Max patterns #2500Reshape/Flattennodes [rewriter] Unify reshape flatten ops #2518PadintoConv/ConvInteger[Rewriter] Add optimizer to fold Pad operators into Conv #2363Clip/ReLUnodes [Rewriter]: fuse successive Relu/Clip nodes #2410We're happy to contribute if any of these are a good fit. Feel free to add other suggestions or let us know if some of these are out of scope.