mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
93408c7cd8
add onnxruntime plugin support, add more custom ops update function_rewriter, enable use origin function add setup.cfg update setup.cfg
12 lines
355 B
C++
12 lines
355 B
C++
#include "nms/trt_nms.hpp"
|
|
#include "roi_align/trt_roi_align.hpp"
|
|
#include "scatternd/trt_scatternd.hpp"
|
|
|
|
REGISTER_TENSORRT_PLUGIN(NonMaxSuppressionDynamicCreator);
|
|
REGISTER_TENSORRT_PLUGIN(ONNXScatterNDDynamicCreator);
|
|
REGISTER_TENSORRT_PLUGIN(RoIAlignPluginDynamicCreator);
|
|
|
|
extern "C" {
|
|
bool initLibMMCVInferPlugins() { return true; }
|
|
} // extern "C"
|