Public Member Functions | |
BatchedNMSBasePluginCreator () | |
~BatchedNMSBasePluginCreator () override=default | |
const char * | getPluginName () const override |
Return the plugin name. More... | |
const char * | getPluginVersion () const override |
Return the plugin version. More... | |
const PluginFieldCollection * | getFieldNames () override |
Return a list of fields that needs to be passed to createPlugin. More... | |
void | setPluginNamespace (const char *libNamespace) override |
Set the namespace of the plugin creator based on the plugin library it belongs to. More... | |
const char * | getPluginNamespace () const override |
Return the namespace of the plugin creator object. More... | |
virtual int32_t | getTensorRTVersion () const |
Return the version of the API the plugin creator was compiled with. More... | |
virtual IPluginV2 * | createPlugin (const char *name, const PluginFieldCollection *fc)=0 |
Return a plugin object. More... | |
virtual IPluginV2 * | deserializePlugin (const char *name, const void *serialData, size_t serialLength)=0 |
Called during deserialization of plugin layer. More... | |
Protected Attributes | |
NMSParameters | params |
bool | mClipBoxes |
std::string | mPluginName |
std::string | mNamespace |
Static Protected Attributes | |
static PluginFieldCollection | mFC {} |
static std::vector< PluginField > | mPluginAttributes |
BatchedNMSBasePluginCreator::BatchedNMSBasePluginCreator | ( | ) |
|
overridedefault |
|
overridevirtual |
Return the plugin name.
Implements nvinfer1::IPluginCreator.
|
overridevirtual |
Return the plugin version.
Implements nvinfer1::IPluginCreator.
|
overridevirtual |
Return a list of fields that needs to be passed to createPlugin.
Implements nvinfer1::IPluginCreator.
|
inlineoverridevirtualinherited |
Set the namespace of the plugin creator based on the plugin library it belongs to.
This can be set while registering the plugin creator.
Implements nvinfer1::IPluginCreator.
|
inlineoverridevirtualinherited |
Return the namespace of the plugin creator object.
Implements nvinfer1::IPluginCreator.
|
inlinevirtualinherited |
Return the version of the API the plugin creator was compiled with.
|
pure virtualinherited |
Return a plugin object.
Return nullptr in case of error.
Implemented in UffPoolPluginV2Creator, nvinfer1::plugin::BatchedNMSDynamicPluginCreator, nvinfer1::plugin::BatchedNMSPluginCreator, nvinfer1::plugin::ProposalPluginCreator, nvinfer1::plugin::MultilevelProposeROIPluginCreator, nvinfer1::plugin::GroupNormalizationPluginCreator, nvinfer1::plugin::ProposalLayerPluginCreator, nvinfer1::plugin::GenerateDetectionPluginCreator, nvinfer1::plugin::DetectionLayerPluginCreator, nvinfer1::plugin::FlattenConcatPluginCreator, nvinfer1::plugin::NormalizePluginCreator, nvinfer1::plugin::CropAndResizePluginCreator, nvinfer1::plugin::MultilevelCropAndResizePluginCreator, nvinfer1::plugin::PyramidROIAlignPluginCreator, nvinfer1::plugin::RPROIPluginCreator, nvinfer1::plugin::GridAnchorBasePluginCreator, nvinfer1::plugin::RegionPluginCreator, nvinfer1::plugin::SpecialSlicePluginCreator, nvinfer1::plugin::PriorBoxPluginCreator, nvinfer1::plugin::ResizeNearestPluginCreator, nvinfer1::plugin::NMSPluginCreator, nvinfer1::plugin::CoordConvACPluginCreator, nvinfer1::plugin::ReorgPluginCreator, nvinfer1::plugin::BatchTilePluginCreator, nvinfer1::plugin::LReluPluginCreator, bert::EmbLayerNormVarSeqlenPluginCreator, nvinfer1::plugin::InstanceNormalizationPluginCreator, bert::QKVToContextInterleavedPluginCreator, bert::SkipLayerNormInterleavedPluginCreator, and nvinfer1::plugin::SplitPluginCreator.
|
pure virtualinherited |
Called during deserialization of plugin layer.
Return a plugin object.
Implemented in UffPoolPluginV2Creator, nvinfer1::plugin::BatchedNMSDynamicPluginCreator, nvinfer1::plugin::BatchedNMSPluginCreator, nvinfer1::plugin::ProposalPluginCreator, nvinfer1::plugin::GroupNormalizationPluginCreator, bert::EmbLayerNormVarSeqlenPluginCreator, nvinfer1::plugin::FlattenConcatPluginCreator, nvinfer1::plugin::NormalizePluginCreator, nvinfer1::plugin::CropAndResizePluginCreator, nvinfer1::plugin::RPROIPluginCreator, nvinfer1::plugin::GridAnchorBasePluginCreator, nvinfer1::plugin::InstanceNormalizationPluginCreator, nvinfer1::plugin::RegionPluginCreator, bert::QKVToContextInterleavedPluginCreator, nvinfer1::plugin::PriorBoxPluginCreator, nvinfer1::plugin::NMSPluginCreator, nvinfer1::plugin::CoordConvACPluginCreator, nvinfer1::plugin::ReorgPluginCreator, bert::SkipLayerNormInterleavedPluginCreator, nvinfer1::plugin::BatchTilePluginCreator, nvinfer1::plugin::LReluPluginCreator, nvinfer1::plugin::MultilevelProposeROIPluginCreator, nvinfer1::plugin::ProposalLayerPluginCreator, nvinfer1::plugin::GenerateDetectionPluginCreator, nvinfer1::plugin::DetectionLayerPluginCreator, nvinfer1::plugin::MultilevelCropAndResizePluginCreator, nvinfer1::plugin::PyramidROIAlignPluginCreator, nvinfer1::plugin::SpecialSlicePluginCreator, nvinfer1::plugin::ResizeNearestPluginCreator, and nvinfer1::plugin::SplitPluginCreator.
|
staticprotected |
|
protected |
|
staticprotected |
|
protected |
|
protected |
|
protectedinherited |