[TOPI][Relay][OP] support dynamic NMS(Non Maximum Suppression), symbolic begin, end...
authorYong Wu <ywu118@alumni.jh.edu>
Tue, 9 Jun 2020 16:48:33 +0000 (00:48 +0800)
committerGitHub <noreply@github.com>
Tue, 9 Jun 2020 16:48:33 +0000 (09:48 -0700)
commitaa808570dbee19a178130c0f7d9b397ff7f51f0e
tree77fa574d32a01357b7099c2ae9783632eb60f557
parent0ea99698f4379eedba26b06a9f426e613bf5b25e
[TOPI][Relay][OP] support dynamic NMS(Non Maximum Suppression), symbolic begin, end, and strides for strided_slice (#4312)

* [TOPI][Relay][OP] Dynamic NMS and strided_slice

* Incorporate comments

* fix nnvm compatibility issues

* fix InferCorrectLayout

* Minor fix

* fix for fuse

* Workaround to pass batch_size into hybrid function to handle dynamic shape

* Seperate rearrange

* fix lint

* fix ci, comments

* change attr to Optional<T>

* clang format

* remove empty lines

* partial ignore for end of strided_slice

* pylint

* add out_indices for gpu get_valid_counts

* change to slice_mode

* clang-format, fix comments

* fix comment

* change slice_mode to string

* fix CI

* update docstring

Co-authored-by: Yao Wang <kevinthesunwy@gmail.com>
43 files changed:
include/tvm/relay/attrs/transform.h
include/tvm/relay/attrs/vision.h
python/tvm/relay/_parser.py
python/tvm/relay/frontend/keras.py
python/tvm/relay/frontend/mxnet.py
python/tvm/relay/frontend/onnx.py
python/tvm/relay/frontend/pytorch.py
python/tvm/relay/frontend/tensorflow.py
python/tvm/relay/frontend/tflite.py
python/tvm/relay/op/_tensor_grad.py
python/tvm/relay/op/_transform.py
python/tvm/relay/op/strategy/generic.py
python/tvm/relay/op/transform.py
python/tvm/relay/op/vision/_vision.py
python/tvm/relay/op/vision/nms.py
src/relay/analysis/util.cc
src/relay/op/tensor/transform.cc
src/relay/op/vision/nms.cc
src/relay/transforms/combine_parallel_conv2d.cc
src/relay/transforms/pattern_util.h
tests/python/frontend/tensorflow/test_debugging.py
tests/python/frontend/tensorflow/test_forward.py
tests/python/relay/test_any.py
tests/python/relay/test_op_level2.py
tests/python/relay/test_op_level4.py
tests/python/relay/test_op_level5.py
tests/python/relay/test_pass_alter_op_layout.py
tests/python/relay/test_pass_combine_parallel_conv2d.py
topi/include/topi/transform.h
topi/python/topi/cuda/conv2d_alter_op.py
topi/python/topi/cuda/nms.py
topi/python/topi/cuda/ssd/multibox.py
topi/python/topi/image/dilation2d.py
topi/python/topi/math.py
topi/python/topi/sort.py
topi/python/topi/testing/strided_slice_python.py
topi/python/topi/transform.py
topi/python/topi/vision/nms.py
topi/python/topi/vision/ssd/multibox.py
topi/python/topi/x86/conv2d_alter_op.py
topi/python/topi/x86/conv3d.py
topi/src/transform.cc
topi/tests/python/test_topi_vision.py