mmdeploy/csrc/backend_ops/tensorrt/batched_rotated_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
..
trt_batched_rotated_nms.cpp [Enhancement] Support RotatedRetinaNet TensorRT (#422) 2022-05-24 10:34:22 +08:00
trt_batched_rotated_nms.hpp [Enhancement] Support RotatedRetinaNet TensorRT (#422) 2022-05-24 10:34:22 +08:00