Skip to content

Commit 2ead004

Browse files
authored
Update batchedNMS plugin to IPluginV2DynamicExt (#738)
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
1 parent 3891af8 commit 2ead004

4 files changed

Lines changed: 372 additions & 58 deletions

File tree

plugin/InferPlugin.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@ bool initLibNvInferPlugins(void* logger, const char* libNamespace)
158158
{
159159
initializePlugin<nvinfer1::plugin::BatchTilePluginCreator>(logger, libNamespace);
160160
initializePlugin<nvinfer1::plugin::BatchedNMSPluginCreator>(logger, libNamespace);
161+
initializePlugin<nvinfer1::plugin::BatchedNMSDynamicPluginCreator>(logger, libNamespace);
161162
initializePlugin<nvinfer1::plugin::CoordConvACPluginCreator>(logger, libNamespace);
162163
initializePlugin<nvinfer1::plugin::CropAndResizePluginCreator>(logger, libNamespace);
163164
initializePlugin<nvinfer1::plugin::DetectionLayerPluginCreator>(logger, libNamespace);

plugin/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
|---|---|---|
77
| [batchTilePlugin](batchTilePlugin) | BatchTilePlugin_TRT | 1 |
88
| [batchedNMSPlugin](batchedNMSPlugin) | BatchedNMS_TRT | 1 |
9+
| [batchedNMSDynamicPlugin](batchedNMSPlugin) | BatchedNMSDynamic_TRT | 1 |
910
| [bertQKVToContextPlugin](bertQKVToContextPlugin) | CustomQKVToContextPluginDynamic | 1 |
1011
| [coordConvACPlugin](coordConvACPlugin) | CoordConvAC | 1 |
1112
| [cropAndResizePlugin](cropAndResizePlugin) | CropAndResize | 1 |

0 commit comments

Comments
 (0)