Improve CUDA conv2d_transpose_nchw (#4762)
authorAlex Gladkov <gladkova@lab126.com>
Wed, 22 Jan 2020 13:41:46 +0000 (05:41 -0800)
committerWuwei Lin <wuwei@apache.org>
Wed, 22 Jan 2020 13:41:46 +0000 (08:41 -0500)
commit4f92cfe5e2f1524285d4111753ad0e24ebc4e318
tree9a24c0e00433a781a141cec38e809a1d947999f7
parentcf3e7865ecfd7ba398f86b887d9d6f75cf0714b0
Improve CUDA conv2d_transpose_nchw (#4762)

- combine pad and dilate;
- fix for the issue https://discuss.tvm.ai/t/compile-error-for-cuda-target/4164
- fix for the issue https://github.com/apache/incubator-tvm/pull/4472
topi/python/topi/cuda/conv2d_transpose_nchw.py
topi/tests/python/test_topi_conv2d_transpose_nchw.py