-
- Downloads
[RELAY] Port winograd ops to relay (#2356)
Showing
- include/tvm/relay/attrs/nn.h 78 additions, 6 deletionsinclude/tvm/relay/attrs/nn.h
- nnvm/python/nnvm/to_relay.py 6 additions, 5 deletionsnnvm/python/nnvm/to_relay.py
- nnvm/python/nnvm/top/attr_dict.py 1 addition, 1 deletionnnvm/python/nnvm/top/attr_dict.py
- nnvm/python/nnvm/top/nn.py 25 additions, 7 deletionsnnvm/python/nnvm/top/nn.py
- python/tvm/attrs.py 39 additions, 0 deletionspython/tvm/attrs.py
- python/tvm/relay/build_module.py 10 additions, 3 deletionspython/tvm/relay/build_module.py
- python/tvm/relay/frontend/mxnet.py 6 additions, 6 deletionspython/tvm/relay/frontend/mxnet.py
- python/tvm/relay/op/nn/_nn.py 54 additions, 6 deletionspython/tvm/relay/op/nn/_nn.py
- python/tvm/relay/op/nn/nn.py 100 additions, 7 deletionspython/tvm/relay/op/nn/nn.py
- python/tvm/relay/op/op_attrs.py 12 additions, 2 deletionspython/tvm/relay/op/op_attrs.py
- python/tvm/relay/testing/__init__.py 2 additions, 0 deletionspython/tvm/relay/testing/__init__.py
- src/relay/op/layout.h 59 additions, 70 deletionssrc/relay/op/layout.h
- src/relay/op/nn/convolution.cc 200 additions, 17 deletionssrc/relay/op/nn/convolution.cc
- src/relay/pass/alter_op_layout.cc 48 additions, 27 deletionssrc/relay/pass/alter_op_layout.cc
- src/relay/pass/combine_parallel_conv2d.cc 5 additions, 5 deletionssrc/relay/pass/combine_parallel_conv2d.cc
- src/relay/pass/fold_scale_axis.cc 22 additions, 28 deletionssrc/relay/pass/fold_scale_axis.cc
- src/relay/pass/pattern_util.h 3 additions, 3 deletionssrc/relay/pass/pattern_util.h
- tests/python/relay/test_op_level2.py 1 addition, 1 deletiontests/python/relay/test_op_level2.py
- tests/python/relay/test_pass_alter_op_layout.py 106 additions, 3 deletionstests/python/relay/test_pass_alter_op_layout.py
- tests/python/relay/test_pass_fold_scale_axis.py 4 additions, 4 deletionstests/python/relay/test_pass_fold_scale_axis.py
Loading
Please register or sign in to comment