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 (279a8eb -> 7e32f37)
Date Wed, 04 Dec 2019 08:44:18 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 279a8eb  [RUNTIME][RPC] Update RPC runtime to allow remote module as arg (#4462)
     add 7e32f37  implement conv3d op (#4400)

No new revisions were added by this update.

Summary of changes:
 include/tvm/relay/attrs/nn.h                       |  56 +++++++++
 python/tvm/relay/op/nn/_nn.py                      |  43 ++++++-
 python/tvm/relay/op/nn/nn.py                       |  85 +++++++++++++
 src/relay/op/nn/convolution.cc                     |  58 +++++++++
 src/relay/op/nn/convolution.h                      | 117 ++++++++++++++++++
 tests/python/relay/test_op_level2.py               |  46 +++++++
 topi/python/topi/cuda/__init__.py                  |   1 +
 topi/python/topi/cuda/conv3d.py                    | 136 +++++++++++++++++++++
 .../cuda/{conv2d_direct.py => conv3d_direct.py}    |  43 ++++---
 topi/python/topi/generic/nn.py                     |  18 +++
 topi/python/topi/nn/__init__.py                    |   1 +
 topi/python/topi/nn/conv3d.py                      | 130 ++++++++++++++++++++
 topi/python/topi/nn/util.py                        |  54 ++++++++
 ...onv2d_winograd.py => test_topi_conv3d_ncdhw.py} |  90 ++++++--------
 14 files changed, 803 insertions(+), 75 deletions(-)
 create mode 100644 topi/python/topi/cuda/conv3d.py
 copy topi/python/topi/cuda/{conv2d_direct.py => conv3d_direct.py} (74%)
 create mode 100644 topi/python/topi/nn/conv3d.py
 copy topi/tests/python/{test_topi_conv2d_winograd.py => test_topi_conv3d_ncdhw.py} (52%)


Mime
View raw message