python/tvm/relay/op/nn/_nn.py · 4ab73634c195fb7006e6279d9dbd71ecba33997b · 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

_nn.py 26.5 KB