projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 2e0dbaa) | patch
[TOPI] Tunable Template for Conv2D HWCN on CUDA (#4168)
authorCody Hao Yu <comaniac0422@gmail.com>
Thu, 24 Oct 2019 19:03:15 +0000 (12:03 -0700)
committerLeyuan Wang <laurawly@gmail.com>
Thu, 24 Oct 2019 19:03:15 +0000 (12:03 -0700)
commit4ab73634c195fb7006e6279d9dbd71ecba33997b
treec7d122c8d74fdddc0c636b8eeb02adb68828235dtree | snapshot
parent2e0dbaa62aadae094bf4819da8211296cd78fe0bcommit | diff
[TOPI] Tunable Template for Conv2D HWCN on CUDA (#4168)

* support conv2d HWCN in AutoTVM and Relay

* fix lint

* fix comments and unit tests
python/tvm/autotvm/task/task.py diff | blob | history
python/tvm/autotvm/task/topi_integration.py diff | blob | history
python/tvm/relay/op/nn/_nn.py diff | blob | history
src/pass/vectorize_loop.cc diff | blob | history
topi/python/topi/cuda/conv2d_hwcn.py diff | blob | history
topi/python/topi/generic/nn.py diff | blob | history
topi/python/topi/nn/conv2d.py diff | blob | history
topi/tests/python/test_topi_conv2d_hwcn.py diff | blob | history
topi/tests/python/test_topi_conv2d_nchw.py diff | blob | history
Domain: Machine Learning / ML Framework;