-
- Downloads
[PASS] PrintGraphIR, SimplifyBatchNormInference (#19)
Showing
- nnvm/include/nnvm/node.h 27 additions, 0 deletionsnnvm/include/nnvm/node.h
- nnvm/python/nnvm/compiler/graph_attr.py 2 additions, 3 deletionsnnvm/python/nnvm/compiler/graph_attr.py
- nnvm/python/nnvm/compiler/graph_pass.py 24 additions, 0 deletionsnnvm/python/nnvm/compiler/graph_pass.py
- nnvm/python/nnvm/graph.py 4 additions, 0 deletionsnnvm/python/nnvm/graph.py
- nnvm/python/nnvm/top/nn.py 2 additions, 3 deletionsnnvm/python/nnvm/top/nn.py
- nnvm/src/compiler/graph_deep_compare.cc 113 additions, 0 deletionsnnvm/src/compiler/graph_deep_compare.cc
- nnvm/src/compiler/graph_fuse.cc 1 addition, 1 deletionnnvm/src/compiler/graph_fuse.cc
- nnvm/src/compiler/graph_transform.h 125 additions, 0 deletionsnnvm/src/compiler/graph_transform.h
- nnvm/src/compiler/layout_transform.cc 0 additions, 0 deletionsnnvm/src/compiler/layout_transform.cc
- nnvm/src/compiler/node_attr.h 34 additions, 0 deletionsnnvm/src/compiler/node_attr.h
- nnvm/src/compiler/packed_func_ext.cc 3 additions, 14 deletionsnnvm/src/compiler/packed_func_ext.cc
- nnvm/src/compiler/precompute_prune.cc 0 additions, 0 deletionsnnvm/src/compiler/precompute_prune.cc
- nnvm/src/compiler/simplify_batch_norm.cc 114 additions, 0 deletionsnnvm/src/compiler/simplify_batch_norm.cc
- nnvm/src/pass/print_graph_ir.cc 127 additions, 0 deletionsnnvm/src/pass/print_graph_ir.cc
- nnvm/src/top/nn/nn_common.h 1 addition, 0 deletionsnnvm/src/top/nn/nn_common.h
- nnvm/src/top/tensor/elemwise.cc 20 additions, 0 deletionsnnvm/src/top/tensor/elemwise.cc
- nnvm/tests/python/compiler/test_graph_pass.py 3 additions, 3 deletionsnnvm/tests/python/compiler/test_graph_pass.py
- nnvm/tests/python/compiler/test_simplify_batchnorm.py 49 additions, 0 deletionsnnvm/tests/python/compiler/test_simplify_batchnorm.py
- nnvm/tests/python/unittest/test_top_level1.py 1 addition, 0 deletionsnnvm/tests/python/unittest/test_top_level1.py
Loading
Please register or sign in to comment