mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmdeploy/backend_ops/tensorrt/grid_sampler
History
RunningLeon 90ce7207da
[Feature] Support Mask R-CNN with ONNX Runtime and TensorRT (#24) 
* add grid_sampler plugin

* update test_mixins to support maskrcnn

* add config for maskrcnn exporting to onnx

* fix lints

* fix lint

* resolve comments

* fix lint

* fix lint

* fix lint

* support trt8

* move mark to single_stage

* fix lint

* remove mark in single stage
 		2021-07-28 19:02:43 +08:00
..
trt_grid_sampler.cpp [Feature] Support Mask R-CNN with ONNX Runtime and TensorRT (#24) 2021-07-28 19:02:43 +08:00
trt_grid_sampler.hpp [Feature] Support Mask R-CNN with ONNX Runtime and TensorRT (#24) 2021-07-28 19:02:43 +08:00
trt_grid_sampler_kernel.cu [Feature] Support Mask R-CNN with ONNX Runtime and TensorRT (#24) 2021-07-28 19:02:43 +08:00
trt_grid_sampler_kernel.hpp [Feature] Support Mask R-CNN with ONNX Runtime and TensorRT (#24) 2021-07-28 19:02:43 +08:00