[TOPI] depthwise-conv2d in NCHW[x]c layout for x86 (#2045)
Showing
- nnvm/python/nnvm/top/nn.py 10 additions, 1 deletionnnvm/python/nnvm/top/nn.py
- nnvm/src/top/nn/convolution.cc 3 additions, 1 deletionnnvm/src/top/nn/convolution.cc
- topi/python/topi/generic/nn.py 17 additions, 0 deletionstopi/python/topi/generic/nn.py
- topi/python/topi/nn/depthwise_conv2d.py 64 additions, 1 deletiontopi/python/topi/nn/depthwise_conv2d.py
- topi/python/topi/x86/__init__.py 1 addition, 0 deletionstopi/python/topi/x86/__init__.py
- topi/python/topi/x86/conv2d.py 65 additions, 49 deletionstopi/python/topi/x86/conv2d.py
- topi/python/topi/x86/conv2d_avx_1x1.py 3 additions, 1 deletiontopi/python/topi/x86/conv2d_avx_1x1.py
- topi/python/topi/x86/conv2d_avx_common.py 3 additions, 1 deletiontopi/python/topi/x86/conv2d_avx_common.py
- topi/python/topi/x86/depthwise_conv2d.py 203 additions, 0 deletionstopi/python/topi/x86/depthwise_conv2d.py
- topi/python/topi/x86/util.py 12 additions, 0 deletionstopi/python/topi/x86/util.py
- topi/tests/python/test_topi_conv2d_NCHWc.py 11 additions, 12 deletionstopi/tests/python/test_topi_conv2d_NCHWc.py
- topi/tests/python/test_topi_depthwise_conv2d.py 116 additions, 0 deletionstopi/tests/python/test_topi_depthwise_conv2d.py
- tutorials/autotvm/tune_nnvm_x86.py 9 additions, 1 deletiontutorials/autotvm/tune_nnvm_x86.py
Loading
Please register or sign in to comment