[RELAY][PASS] FoldScaleAxis Forward (#2020)
* [RELAY][PASS] FoldScaleAxis Forward * Introduce helper function type_as * Update per review comment * Fix according to comments
Showing
- include/tvm/relay/attrs/transform.h 7 additions, 6 deletionsinclude/tvm/relay/attrs/transform.h
- include/tvm/relay/expr.h 26 additions, 0 deletionsinclude/tvm/relay/expr.h
- include/tvm/relay/expr_functor.h 11 additions, 3 deletionsinclude/tvm/relay/expr_functor.h
- include/tvm/relay/op.h 27 additions, 16 deletionsinclude/tvm/relay/op.h
- python/tvm/relay/ir_pass.py 20 additions, 0 deletionspython/tvm/relay/ir_pass.py
- python/tvm/relay/op/transform.py 8 additions, 10 deletionspython/tvm/relay/op/transform.py
- src/relay/ir/alpha_equal.cc 15 additions, 5 deletionssrc/relay/ir/alpha_equal.cc
- src/relay/ir/expr_functor.cc 2 additions, 2 deletionssrc/relay/ir/expr_functor.cc
- src/relay/op/tensor/transform.cc 6 additions, 8 deletionssrc/relay/op/tensor/transform.cc
- src/relay/pass/fold_scale_axis.cc 554 additions, 0 deletionssrc/relay/pass/fold_scale_axis.cc
- src/relay/pass/pattern_util.h 123 additions, 0 deletionssrc/relay/pass/pattern_util.h
- src/relay/pass/type_infer.cc 45 additions, 13 deletionssrc/relay/pass/type_infer.cc
- tests/python/relay/test_op_level3.py 4 additions, 4 deletionstests/python/relay/test_op_level3.py
- tests/python/relay/test_pass_fold_scale_axis.py 153 additions, 0 deletionstests/python/relay/test_pass_fold_scale_axis.py
Loading
Please register or sign in to comment