mmdeploy/csrc/backend_ops/tensorrt/common/nms
q.yao e3a8baac4c
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
* add rotated nms trt plugin

* fix ops output shape

* rebase

* fix lint

* add fp16, benchmark

* format docs

* remove unused definition, add ut

* add docs

* update docs

* add doc
2022-05-24 10:34:22 +08:00
..
batched_nms_kernel.hpp [Enhancement] Support RotatedRetinaNet TensorRT (#422) 2022-05-24 10:34:22 +08:00
cub_helper.h [Enhancement] Support RotatedRetinaNet TensorRT (#422) 2022-05-24 10:34:22 +08:00
kernel.h [Enhancement] Support RotatedRetinaNet TensorRT (#422) 2022-05-24 10:34:22 +08:00