You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by mb...@apache.org on 2021/05/04 16:26:52 UTC
[tvm] branch main updated (3ed83b6 -> 38e0bbe)
This is an automated email from the ASF dual-hosted git repository.
mbrookhart pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 3ed83b6 [Relay][Autoscheduler] Fix autoscheduler matmul without units. (#7957)
add 38e0bbe [ONNX][TOPI][Relay]Support dilations in pooling operators (#7928)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/nn.h | 24 +
include/tvm/topi/nn/pooling.h | 292 +++-----
python/tvm/relay/frontend/onnx.py | 30 +-
python/tvm/relay/frontend/pytorch.py | 60 +-
python/tvm/relay/op/nn/nn.py | 79 ++-
python/tvm/topi/nn/pooling.py | 143 ++--
python/tvm/topi/testing/__init__.py | 3 +-
python/tvm/topi/testing/pool1d_python.py | 69 --
python/tvm/topi/testing/pool3d_python.py | 111 ---
python/tvm/topi/testing/poolnd_python.py | 161 +++++
src/relay/op/nn/pooling.cc | 126 ++--
src/relay/op/nn/pooling.h | 9 +-
src/relay/qnn/op/convolution.cc | 32 +-
src/relay/transforms/pattern_utils.h | 6 +-
src/topi/nn.cc | 18 +-
.../contrib/test_arm_compute_lib/test_pooling.py | 59 +-
tests/python/frontend/onnx/test_forward.py | 18 +-
tests/python/frontend/pytorch/test_forward.py | 16 +-
tests/python/relay/test_any.py | 26 +-
tests/python/relay/test_op_level2.py | 230 +++---
tests/python/relay/test_op_level5.py | 8 +-
tests/python/topi/python/test_topi_pooling.py | 777 ++++++++++++++++-----
.../python/unittest/test_auto_scheduler_common.py | 5 +-
tests/python/unittest/test_custom_datatypes.py | 15 +-
tests/python/unittest/test_te_autodiff.py | 30 +-
tests/python/unittest/test_te_tensor.py | 8 +-
26 files changed, 1419 insertions(+), 936 deletions(-)
delete mode 100644 python/tvm/topi/testing/pool1d_python.py
delete mode 100644 python/tvm/topi/testing/pool3d_python.py
create mode 100644 python/tvm/topi/testing/poolnd_python.py