Skip to content

Commit

Permalink
[ONNX] rework some reduction op lowerings (#3870)
Browse files Browse the repository at this point in the history
- Refactors more "onnx.ReduceXXX" patterns through helper function.
- Fixes bug with iterating unconditionally on `output_dim == 1` during
`dimList` inference.

This change results in passes for the following 11 models:

crossvit_15_240
crossvit_15_dagger_240
crossvit_15_dagger_408
crossvit_18_240
crossvit_18_dagger_240
crossvit_18_dagger_408
crossvit_9_240
crossvit_9_dagger_240
crossvit_base_240
crossvit_small_240
crossvit_tiny_240

---------

Co-authored-by: Vinayak Dev <104419489+vinayakdsci@users.noreply.github.com>
  • Loading branch information
zjgarvey and vinayakdsci authored Nov 14, 2024
1 parent 30c5193 commit 1201bab
Show file tree
Hide file tree
Showing 2 changed files with 186 additions and 468 deletions.
Loading

0 comments on commit 1201bab

Please sign in to comment.