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