-
- Downloads
[TOP] Add dense, batchnorm (#22)
* [TOP] Add dense, batchnorm * update tvm
Showing
- nnvm/include/nnvm/compiler/op_attr_types.h 5 additions, 2 deletionsnnvm/include/nnvm/compiler/op_attr_types.h
- nnvm/python/nnvm/compiler/build_module.py 11 additions, 2 deletionsnnvm/python/nnvm/compiler/build_module.py
- nnvm/python/nnvm/compiler/registry.py 3 additions, 1 deletionnnvm/python/nnvm/compiler/registry.py
- nnvm/python/nnvm/top/__init__.py 1 addition, 0 deletionsnnvm/python/nnvm/top/__init__.py
- nnvm/python/nnvm/top/nn.py 39 additions, 10 deletionsnnvm/python/nnvm/top/nn.py
- nnvm/python/nnvm/top/tensor.py 93 additions, 28 deletionsnnvm/python/nnvm/top/tensor.py
- nnvm/python/nnvm/top/transform.py 31 additions, 0 deletionsnnvm/python/nnvm/top/transform.py
- nnvm/src/compiler/graph_fuse.cc 14 additions, 3 deletionsnnvm/src/compiler/graph_fuse.cc
- nnvm/src/compiler/packed_func_ext.cc 4 additions, 2 deletionsnnvm/src/compiler/packed_func_ext.cc
- nnvm/src/compiler/simplify_batch_norm.cc 6 additions, 13 deletionsnnvm/src/compiler/simplify_batch_norm.cc
- nnvm/src/pass/print_graph_ir.cc 1 addition, 1 deletionnnvm/src/pass/print_graph_ir.cc
- nnvm/tests/python/compiler/test_simplify_batchnorm.py 4 additions, 5 deletionsnnvm/tests/python/compiler/test_simplify_batchnorm.py
- nnvm/tests/python/compiler/test_top_level1.py 150 additions, 99 deletionsnnvm/tests/python/compiler/test_top_level1.py
- nnvm/tests/python/compiler/test_top_level2.py 39 additions, 47 deletionsnnvm/tests/python/compiler/test_top_level2.py
Loading
Please register or sign in to comment