[PASS] FoldScaleAxis (#55)
* [PASS] FoldScaleAxis * Move FoldAxis to O3 * Set unroll to 0 when ready
Showing
- nnvm/include/nnvm/graph.h 2 additions, 0 deletionsnnvm/include/nnvm/graph.h
- nnvm/python/nnvm/compiler/build_module.py 8 additions, 2 deletionsnnvm/python/nnvm/compiler/build_module.py
- nnvm/python/nnvm/testing/init.py 0 additions, 1 deletionnnvm/python/nnvm/testing/init.py
- nnvm/python/nnvm/testing/mobilenet.py 2 additions, 1 deletionnnvm/python/nnvm/testing/mobilenet.py
- nnvm/python/nnvm/testing/utils.py 1 addition, 1 deletionnnvm/python/nnvm/testing/utils.py
- nnvm/src/README.md 3 additions, 4 deletionsnnvm/src/README.md
- nnvm/src/compiler/compile_engine.cc 1 addition, 1 deletionnnvm/src/compiler/compile_engine.cc
- nnvm/src/compiler/fold_scale_axis.cc 271 additions, 0 deletionsnnvm/src/compiler/fold_scale_axis.cc
- nnvm/src/compiler/graph_fuse.cc 3 additions, 9 deletionsnnvm/src/compiler/graph_fuse.cc
- nnvm/src/compiler/graph_transform.h 3 additions, 4 deletionsnnvm/src/compiler/graph_transform.h
- nnvm/src/compiler/pattern_util.h 99 additions, 0 deletionsnnvm/src/compiler/pattern_util.h
- nnvm/src/compiler/simplify_inference.cc 6 additions, 11 deletionsnnvm/src/compiler/simplify_inference.cc
- nnvm/src/core/graph.cc 1 addition, 0 deletionsnnvm/src/core/graph.cc
- nnvm/src/top/tensor/transform.cc 4 additions, 4 deletionsnnvm/src/top/tensor/transform.cc
- nnvm/tests/python/compiler/test_fold_axis.py 49 additions, 0 deletionsnnvm/tests/python/compiler/test_fold_axis.py
- nnvm/tests/python/compiler/test_simplify_inference.py 2 additions, 4 deletionsnnvm/tests/python/compiler/test_simplify_inference.py
- nnvm/tests/python/frontend/mxnet/test_forward.py 2 additions, 1 deletionnnvm/tests/python/frontend/mxnet/test_forward.py
Loading
Please register or sign in to comment