projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 8781d1f) | patch
[RELAY][DYN] Dynamic upsampling relay op (#6273)
authorLily Orth-Smith <lilyorthsmith@gmail.com>
Fri, 21 Aug 2020 16:06:53 +0000 (09:06 -0700)
committerGitHub <noreply@github.com>
Fri, 21 Aug 2020 16:06:53 +0000 (09:06 -0700)
commit158e9beeb547699727b164db037bb21a22d47bea
treebb08911e4259a382398ec481a9213e04963dfdedtree | snapshot
parent8781d1f565e674ef85b4972eb3749f2ccebaba04commit | diff
[RELAY][DYN] Dynamic upsampling relay op (#6273)

* implementing upsampling op

* fix lint

* fix lint again

* add doc to upsampling shape func

* fix set attrs build problem

* fixing imports

* reverting data layout transform changes

* moved layout template to header file

* changing python module from nn.dyn to dyn.nn

* adding support for more layouts to upsampling

* fix lint

* fix upsampling doc

* change _nn.py doc

* failed flakey test

* fix build after merge
python/tvm/relay/op/dyn/nn/_nn.py diff | blob | history
python/tvm/relay/op/nn/nn.py diff | blob | history
python/tvm/topi/nn/upsampling.py diff | blob | history
src/relay/op/dyn/nn/upsampling.cc [new file with mode: 0644] blob
src/relay/op/make_op.h diff | blob | history
src/relay/op/nn/upsampling.cc diff | blob | history
src/relay/op/nn/upsampling.h [new file with mode: 0644] blob
src/relay/transforms/dynamic_to_static.cc diff | blob | history
tests/python/relay/dyn/test_dynamic_op_level2.py diff | blob | history
tests/python/relay/test_pass_dynamic_to_static.py diff | blob | history
Domain: Machine Learning / ML Framework;