projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 028f47c) | patch
[VTA][TOPI] Conv2d transpose (deconvolution) operator support (#3777)
authorThierry Moreau <moreau@uw.edu>
Thu, 5 Sep 2019 18:29:42 +0000 (11:29 -0700)
committerJared Roesch <roeschinc@gmail.com>
Thu, 5 Sep 2019 18:29:42 +0000 (11:29 -0700)
commit23c22812b8e2a46133cfbc89e4910e24dc0b976a
tree854c17d7d7424913bbfa878567870480ab97ed4etree | snapshot
parent028f47ce654d7419e0cebb274541696e960fb90ccommit | diff
[VTA][TOPI] Conv2d transpose (deconvolution) operator support (#3777)

* initial conv2d_transpose

* correct select operator

* cleanup

* fix

* fix correcness check

* conv2d transpose declaration fix

* autotvm conv2d_transpose tuning script

* ir pass fix

* fix tuning script

* deriving params from env, adding bias

* removing bias comp from deconvolution

* lint

* fix

* lint

* lint

* turning off cpu

* lint, ops

* lint

* import fix

* removing hard coded values

* lint
vta/python/vta/build_module.py diff | blob | history
vta/python/vta/ir_pass.py diff | blob | history
vta/python/vta/top/__init__.py diff | blob | history
vta/python/vta/top/op.py diff | blob | history
vta/python/vta/top/util.py [new file with mode: 0644] blob
vta/python/vta/top/vta_conv2d.py diff | blob | history
vta/python/vta/top/vta_conv2d_transpose.py [new file with mode: 0644] blob
vta/scripts/tune_conv2d.py diff | blob | history
vta/scripts/tune_conv2d_transpose.py [new file with mode: 0644] blob
vta/tests/python/integration/test_benchmark_topi_conv2d.py diff | blob | history
vta/tests/python/integration/test_benchmark_topi_conv2d_transpose.py [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;