[TOPI][RELAY][PYTORCH]Conv3d_transpose op support added (#5737)
authorSamuel <siju.samuel@huawei.com>
Thu, 11 Jun 2020 09:12:35 +0000 (14:42 +0530)
committerGitHub <noreply@github.com>
Thu, 11 Jun 2020 09:12:35 +0000 (18:12 +0900)
commite2fb5039a2472403f2f6dfd65ac6b94de6758271
tree35cec47b78ce066e4814ee31e5d9a022242551d4
parentc7274fd3b0f693fc6214a450e13d5e99026337ae
[TOPI][RELAY][PYTORCH]Conv3d_transpose op support added (#5737)

* [TOPI][RELAY][PYTORCH]Conv3d_transpose op support added

* Test cases in topi/relay

* conv3d_transpose_ncdhw_python added

* Review comments fixed
25 files changed:
docs/api/python/topi.rst
docs/langref/relay_op.rst
include/tvm/relay/attrs/nn.h
python/tvm/relay/frontend/pytorch.py
python/tvm/relay/op/nn/_nn.py
python/tvm/relay/op/nn/nn.py
python/tvm/relay/op/op_attrs.py
python/tvm/relay/op/strategy/cuda.py
python/tvm/relay/op/strategy/generic.py
python/tvm/relay/op/strategy/x86.py
src/relay/op/nn/convolution.cc
src/relay/op/nn/convolution.h
tests/python/frontend/pytorch/test_forward.py
tests/python/relay/test_op_level2.py
topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/conv3d.py
topi/python/topi/cuda/conv3d_transpose_ncdhw.py [new file with mode: 0644]
topi/python/topi/generic/nn.py
topi/python/topi/nn/__init__.py
topi/python/topi/nn/conv3d_transpose.py [new file with mode: 0644]
topi/python/topi/testing/__init__.py
topi/python/topi/testing/conv3d_transpose_ncdhw_python.py [new file with mode: 0644]
topi/python/topi/x86/__init__.py
topi/python/topi/x86/conv3d_transpose.py [new file with mode: 0644]
topi/tests/python/test_topi_conv3d_transpose_ncdhw.py [new file with mode: 0644]