Improve non_max_suppression and get_valid_counts for CPU (#3305)
authorYao Wang <kevinthesunwy@gmail.com>
Sun, 9 Jun 2019 20:34:56 +0000 (13:34 -0700)
committerWuwei Lin <vincentl13x@gmail.com>
Sun, 9 Jun 2019 20:34:56 +0000 (22:34 +0200)
commit98a91af993e6b60a72ce288a7e6f98a44126a918
treeac8faee4b4eff0434e6bda79cc29be89d584356b
parenta4bc50ebfffe034490330a781ac077d958d43286
Improve non_max_suppression and get_valid_counts for CPU (#3305)

* Improve non_max_suppression for CPU

* Improve get_valid_counts

* Minor change

* Skip some unnecessary computes
include/tvm/relay/attrs/vision.h
python/tvm/relay/frontend/mxnet.py
python/tvm/relay/op/vision/_vision.py
python/tvm/relay/op/vision/nms.py
src/relay/op/vision/nms.cc
tests/python/relay/test_op_level5.py
topi/python/topi/cuda/nms.py
topi/python/topi/vision/nms.py
topi/tests/python/test_topi_vision.py