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 2021/04/14 19:19:14 UTC
[tvm] branch main updated (303a2f0 -> 390b4d1)
This is an automated email from the ASF dual-hosted git repository.
masahi pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 303a2f0 [BugFix] Print doubles with precision 17 in SaveJSON and TVM script printer (#7846)
add 390b4d1 [TOPI, Relay] A new NMS op variant for ONNX NMS / TF Combined NMS (#7796)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/vision.h | 11 +-
python/tvm/relay/frontend/onnx.py | 238 +--------------
python/tvm/relay/op/op_attrs.py | 5 +
python/tvm/relay/op/strategy/cuda.py | 12 +
python/tvm/relay/op/strategy/generic.py | 28 +-
python/tvm/relay/op/vision/_vision.py | 19 ++
python/tvm/relay/op/vision/nms.py | 48 +++
python/tvm/topi/cuda/__init__.py | 2 +-
python/tvm/topi/cuda/nms.py | 423 ++++++++++++++++++---------
python/tvm/topi/cuda/scan.py | 4 +-
python/tvm/topi/cuda/sort.py | 27 +-
python/tvm/topi/cuda/vision.py | 2 +-
python/tvm/topi/vision/nms.py | 193 +++++++++++-
python/tvm/topi/vision/nms_util.py | 282 ++++++++++++++++++
src/relay/op/vision/nms.cc | 63 ++++
tests/python/frontend/onnx/test_forward.py | 2 +-
tests/python/relay/test_op_level5.py | 97 +++++-
tests/python/topi/python/test_topi_vision.py | 112 +++++++
18 files changed, 1171 insertions(+), 397 deletions(-)
create mode 100644 python/tvm/topi/vision/nms_util.py