You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2020/06/11 09:12:51 UTC
[incubator-tvm] branch master updated (c7274fd -> e2fb503)
This is an automated email from the ASF dual-hosted git repository.
masahi pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from c7274fd [Relay] Fix for recursive let (#5757)
add e2fb503 [TOPI][RELAY][PYTORCH]Conv3d_transpose op support added (#5737)
No new revisions were added by this update.
Summary of changes:
docs/api/python/topi.rst | 4 +
docs/langref/relay_op.rst | 4 +-
include/tvm/relay/attrs/nn.h | 76 +++++++++
python/tvm/relay/frontend/pytorch.py | 21 ++-
python/tvm/relay/op/nn/_nn.py | 25 +++
python/tvm/relay/op/nn/nn.py | 70 +++++++++
python/tvm/relay/op/op_attrs.py | 3 +
python/tvm/relay/op/strategy/cuda.py | 18 +++
python/tvm/relay/op/strategy/generic.py | 38 +++++
python/tvm/relay/op/strategy/x86.py | 18 +++
src/relay/op/nn/convolution.cc | 45 ++++++
src/relay/op/nn/convolution.h | 97 ++++++++++++
tests/python/frontend/pytorch/test_forward.py | 28 ++++
tests/python/relay/test_op_level2.py | 62 ++++++++
topi/python/topi/cuda/__init__.py | 1 +
topi/python/topi/cuda/conv3d.py | 2 +-
topi/python/topi/cuda/conv3d_transpose_ncdhw.py | 145 ++++++++++++++++++
topi/python/topi/generic/nn.py | 18 +++
topi/python/topi/nn/__init__.py | 1 +
topi/python/topi/nn/conv3d_transpose.py | 169 +++++++++++++++++++++
topi/python/topi/testing/__init__.py | 1 +
.../topi/testing/conv3d_transpose_ncdhw_python.py | 86 +++++++++++
topi/python/topi/x86/__init__.py | 1 +
.../{conv2d_transpose.py => conv3d_transpose.py} | 23 +--
...nchw.py => test_topi_conv3d_transpose_ncdhw.py} | 62 ++++----
25 files changed, 967 insertions(+), 51 deletions(-)
create mode 100644 topi/python/topi/cuda/conv3d_transpose_ncdhw.py
create mode 100644 topi/python/topi/nn/conv3d_transpose.py
create mode 100644 topi/python/topi/testing/conv3d_transpose_ncdhw_python.py
copy topi/python/topi/x86/{conv2d_transpose.py => conv3d_transpose.py} (74%)
copy topi/tests/python/{test_topi_conv2d_transpose_nchw.py => test_topi_conv3d_transpose_ncdhw.py} (50%)