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 2019/12/04 08:44:18 UTC
[incubator-tvm] branch master updated (279a8eb -> 7e32f37)
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%)