[RELAY][PASS] FoldScaleAxis Backward (#2024) (d5103bbc) · Commits · cld / ml / tvm · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

tvm
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents
- Service Desk

Snippets Groups Projects

- cld
- ml
- tvm
- Commits
d5103bbc

Unverified Commit d5103bbc authored 6 years ago by Tianqi Chen Committed by GitHub 6 years ago

Downloads
- Patches
- Plain Diff

[RELAY][PASS] FoldScaleAxis Backward (#2024)

parent 25e4dc51

No related branches found

No related tags found

No related merge requests found

Changes 6

Hide whitespace changes

Inline Side-by-side

Showing

include/tvm/relay/expr_functor.h 3 additions, 3 deletions

include/tvm/relay/expr_functor.h
python/tvm/relay/ir_pass.py 29 additions, 0 deletions

python/tvm/relay/ir_pass.py
src/relay/ir/expr_functor.cc 8 additions, 4 deletions

src/relay/ir/expr_functor.cc
src/relay/pass/fold_scale_axis.cc 433 additions, 22 deletions

src/relay/pass/fold_scale_axis.cc
src/relay/pass/pattern_util.h 21 additions, 2 deletions

src/relay/pass/pattern_util.h
tests/python/relay/test_pass_fold_scale_axis.py 173 additions, 4 deletions

tests/python/relay/test_pass_fold_scale_axis.py

with 667 additions and 35 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment