topi/tests/python/test_topi_conv2d_nchw.py · 6ab158063d53f9dcc992777dc7f2b72d2092e802 · wenyuanbo / tic · GitLab

Blame History Permalink

[TOPI] Tunable Template for Conv2D HWCN on CUDA (#4168) · 4ab73634
```
* support conv2d HWCN in AutoTVM and Relay

* fix lint

* fix comments and unit tests
```
Cody Hao Yu committed Oct 24, 2019
4ab73634

test_topi_conv2d_nchw.py 7.64 KB