mmdeploy/tests/test_ops
q.yao 0caeaf238c
[CustomOps] TensorRT Gather Topk Ops (#1033)
* add gather topk

* add shape inference and document

* fix faster rcnn

* reshape topk

* fix
2022-09-19 13:48:26 +08:00
..
__init__.py Add licence head to code (#236) 2021-11-30 15:00:37 +08:00
test_ops.py [CustomOps] TensorRT Gather Topk Ops (#1033) 2022-09-19 13:48:26 +08:00
utils.py [Enhancement] TensorRT Anchor generator plugin (#646) 2022-06-28 16:11:24 +08:00