[ONNX] NMS in ONNX#6839

Merged

zhiics merged 10 commits into

apache:mainfrom

mbrookhart:mbrookhart/onnx_nms

Dec 14, 2020

Commits on Dec 13, 2020

NMS partially working on CPU, fails on GPU
Matthew Brookhart
committed
support dynamic iou_threshold
Matthew Brookhart
committed
WIP NMS with while loops
Matthew Brookhart
committed
working nms with dynamic shapes
Matthew Brookhart
committed
add a test with dynamic score_threshold and pass it
Matthew Brookhart
committed
Fix type checking in lambda lift

jroesch
authored and
mbrookhart
committed
ONNX NMS working on GPU, had to remove threading from some kernels
Matthew Brookhart
committed
better parallelize get_valid_counts
mbrookhart
committed
improve nms parallelization
mbrookhart
committed
respond to cuda/thrust enablement issue
mbrookhart
committed