General Layout Support (#447)
Showing
- nnvm/include/nnvm/compiler/contrib_op_param.h 0 additions, 28 deletionsnnvm/include/nnvm/compiler/contrib_op_param.h
- nnvm/include/nnvm/compiler/op_attr_types.h 10 additions, 11 deletionsnnvm/include/nnvm/compiler/op_attr_types.h
- nnvm/include/nnvm/compiler/packed_func_ext.h 2 additions, 0 deletionsnnvm/include/nnvm/compiler/packed_func_ext.h
- nnvm/include/nnvm/graph_attr_types.h 17 additions, 1 deletionnnvm/include/nnvm/graph_attr_types.h
- nnvm/include/nnvm/layout.h 455 additions, 0 deletionsnnvm/include/nnvm/layout.h
- nnvm/include/nnvm/op_attr_types.h 26 additions, 0 deletionsnnvm/include/nnvm/op_attr_types.h
- nnvm/include/nnvm/top/nn.h 41 additions, 39 deletionsnnvm/include/nnvm/top/nn.h
- nnvm/python/nnvm/_ctypes/symbol.py 4 additions, 1 deletionnnvm/python/nnvm/_ctypes/symbol.py
- nnvm/python/nnvm/compiler/build_module.py 30 additions, 6 deletionsnnvm/python/nnvm/compiler/build_module.py
- nnvm/python/nnvm/compiler/graph_attr.py 15 additions, 4 deletionsnnvm/python/nnvm/compiler/graph_attr.py
- nnvm/python/nnvm/contrib.py 1 addition, 0 deletionsnnvm/python/nnvm/contrib.py
- nnvm/python/nnvm/frontend/mxnet.py 11 additions, 1 deletionnnvm/python/nnvm/frontend/mxnet.py
- nnvm/python/nnvm/symbol.py 1 addition, 0 deletionsnnvm/python/nnvm/symbol.py
- nnvm/python/nnvm/top/nn.py 42 additions, 1 deletionnnvm/python/nnvm/top/nn.py
- nnvm/python/nnvm/top/registry.py 27 additions, 0 deletionsnnvm/python/nnvm/top/registry.py
- nnvm/src/c_api/c_api_symbolic.cc 1 addition, 1 deletionnnvm/src/c_api/c_api_symbolic.cc
- nnvm/src/compiler/alter_op_layout.cc 151 additions, 0 deletionsnnvm/src/compiler/alter_op_layout.cc
- nnvm/src/compiler/fold_scale_axis.cc 4 additions, 4 deletionsnnvm/src/compiler/fold_scale_axis.cc
- nnvm/src/compiler/layout_transform.cc 0 additions, 159 deletionsnnvm/src/compiler/layout_transform.cc
- nnvm/src/compiler/packed_func_ext.cc 19 additions, 1 deletionnnvm/src/compiler/packed_func_ext.cc
Loading
Please register or sign in to comment