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/01/09 09:43:37 UTC
[incubator-tvm] branch master updated (baae28b -> 8a98a2e)
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 baae28b [Autotvm] Use VM compile to extract autotvm tasks (#4328)
add 8a98a2e [Relay/Topi][Op] 1D Pooling (#4663)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/nn.h | 60 ++++++++++
python/tvm/relay/frontend/onnx.py | 60 +++++++---
python/tvm/relay/op/nn/_nn.py | 24 ++++
python/tvm/relay/op/nn/nn.py | 105 ++++++++++++++++-
python/tvm/relay/op/op_attrs.py | 10 ++
src/relay/op/nn/pooling.cc | 178 +++++++++++++++++++++++++++++
tests/python/frontend/onnx/test_forward.py | 95 +++++++++++++++
tests/python/relay/test_op_level2.py | 29 +++++
topi/include/topi/nn/pooling.h | 55 +++++++++
topi/python/topi/nn/pooling.py | 61 ++++++++++
topi/python/topi/testing/__init__.py | 1 +
topi/python/topi/testing/pool1d_python.py | 66 +++++++++++
topi/src/topi.cc | 7 ++
topi/tests/python/test_topi_pooling.py | 53 +++++++++
14 files changed, 786 insertions(+), 18 deletions(-)
create mode 100644 topi/python/topi/testing/pool1d_python.py