tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From mas...@apache.org
Subject [incubator-tvm] branch master updated (c7274fd -> e2fb503)
Date Thu, 11 Jun 2020 09:12:51 GMT
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%)


Mime
View raw message