-
- Downloads
"README.md" did not exist on "8af06c17fd1c346583e99bdf9f7dab5d8262161c"
[TOPI] support dilated conv2d and depthwise_conv2d (#1129)
* support dilation in conv2d and depthwise_conv2d * handle dilated conv in extern libs (cudnn, miopen)
Showing
- topi/python/topi/cuda/conv2d.py 23 additions, 5 deletionstopi/python/topi/cuda/conv2d.py
- topi/python/topi/cuda/conv2d_hwcn.py 2 additions, 0 deletionstopi/python/topi/cuda/conv2d_hwcn.py
- topi/python/topi/cuda/conv2d_nchw.py 2 additions, 0 deletionstopi/python/topi/cuda/conv2d_nchw.py
- topi/python/topi/cuda/depthwise_conv2d.py 4 additions, 0 deletionstopi/python/topi/cuda/depthwise_conv2d.py
- topi/python/topi/generic/extern.py 1 addition, 1 deletiontopi/python/topi/generic/extern.py
- topi/python/topi/mali/conv2d.py 12 additions, 0 deletionstopi/python/topi/mali/conv2d.py
- topi/python/topi/mali/depthwise_conv2d.py 2 additions, 0 deletionstopi/python/topi/mali/depthwise_conv2d.py
- topi/python/topi/nn/dilate.py 1 addition, 1 deletiontopi/python/topi/nn/dilate.py
- topi/python/topi/nn/pad.py 1 addition, 1 deletiontopi/python/topi/nn/pad.py
- topi/python/topi/opengl/conv2d_nchw.py 3 additions, 0 deletionstopi/python/topi/opengl/conv2d_nchw.py
- topi/python/topi/rasp/conv2d.py 2 additions, 0 deletionstopi/python/topi/rasp/conv2d.py
- topi/python/topi/rasp/depthwise_conv2d.py 2 additions, 0 deletionstopi/python/topi/rasp/depthwise_conv2d.py
- topi/python/topi/rocm/conv2d.py 17 additions, 4 deletionstopi/python/topi/rocm/conv2d.py
- topi/python/topi/x86/conv2d.py 7 additions, 0 deletionstopi/python/topi/x86/conv2d.py
- topi/tests/python/test_topi_conv2d_hwcn.py 7 additions, 4 deletionstopi/tests/python/test_topi_conv2d_hwcn.py
- topi/tests/python/test_topi_conv2d_nchw.py 11 additions, 6 deletionstopi/tests/python/test_topi_conv2d_nchw.py
- topi/tests/python/test_topi_conv2d_nhwc.py 7 additions, 3 deletionstopi/tests/python/test_topi_conv2d_nhwc.py
- topi/tests/python/test_topi_depthwise_conv2d.py 14 additions, 7 deletionstopi/tests/python/test_topi_depthwise_conv2d.py
Loading
Please register or sign in to comment