-
- Downloads
[TOP][COMPILER] Add expand_dims, change graph_compare to not compare input optionally (#25)
Showing
- nnvm/include/nnvm/top/tensor.h 11 additions, 0 deletionsnnvm/include/nnvm/top/tensor.h
- nnvm/python/nnvm/compiler/build_module.py 2 additions, 0 deletionsnnvm/python/nnvm/compiler/build_module.py
- nnvm/python/nnvm/compiler/graph_util.py 7 additions, 2 deletionsnnvm/python/nnvm/compiler/graph_util.py
- nnvm/python/nnvm/testing/__init__.py 1 addition, 0 deletionsnnvm/python/nnvm/testing/__init__.py
- nnvm/python/nnvm/testing/config.py 12 additions, 0 deletionsnnvm/python/nnvm/testing/config.py
- nnvm/python/nnvm/top/nn.py 1 addition, 1 deletionnnvm/python/nnvm/top/nn.py
- nnvm/python/nnvm/top/tensor.py 12 additions, 0 deletionsnnvm/python/nnvm/top/tensor.py
- nnvm/python/nnvm/top/transform.py 10 additions, 0 deletionsnnvm/python/nnvm/top/transform.py
- nnvm/src/compiler/graph_deep_compare.cc 8 additions, 2 deletionsnnvm/src/compiler/graph_deep_compare.cc
- nnvm/src/compiler/simplify_inference.cc 8 additions, 9 deletionsnnvm/src/compiler/simplify_inference.cc
- nnvm/src/top/tensor/transform.cc 44 additions, 1 deletionnnvm/src/top/tensor/transform.cc
- nnvm/tests/python/compiler/test_build.py 0 additions, 3 deletionsnnvm/tests/python/compiler/test_build.py
- nnvm/tests/python/compiler/test_simplify_inference.py 6 additions, 3 deletionsnnvm/tests/python/compiler/test_simplify_inference.py
- nnvm/tests/python/compiler/test_top_level1.py 8 additions, 13 deletionsnnvm/tests/python/compiler/test_top_level1.py
- nnvm/tests/python/compiler/test_top_level2.py 8 additions, 14 deletionsnnvm/tests/python/compiler/test_top_level2.py
- nnvm/tests/python/unittest/test_infer_shape.py 11 additions, 0 deletionsnnvm/tests/python/unittest/test_infer_shape.py
- nnvm/tests/python/unittest/test_top_level1.py 6 additions, 0 deletionsnnvm/tests/python/unittest/test_top_level1.py
Loading
Please register or sign in to comment