TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
PluginFactory Class Referenceabstract
Inheritance diagram for PluginFactory:
Collaboration diagram for PluginFactory:

Public Member Functions

bool isPlugin (const char *name) override
 A user implemented function that determines if a layer configuration is provided by an IPlugin. More...
 
bool isPluginExt (const char *name) override
 A user implemented function that determines if a layer configuration is provided by an IPluginExt. More...
 
virtual IPlugin * createPlugin (const char *layerName, const nvinfer1::Weights *weights, int nbWeights) override
 
nvinfer1::IPlugincreatePlugin (const char *layerName, const void *serialData, size_t serialLength) override
 Create a plugin from serialized data. More...
 
void destroyPlugin ()
 
virtual int32_t getVersion () const
 
virtual nvinfer1::IPlugincreatePlugin (const char *layerName, const nvinfer1::Weights *weights, int32_t nbWeights)=0
 Creates a plugin. More...
 

Public Attributes

std::unique_ptr< FCPluginmPlugin {nullptr}
 

Member Function Documentation

◆ isPlugin()

bool PluginFactory::isPlugin ( const char *  layerName)
inlineoverridevirtual

A user implemented function that determines if a layer configuration is provided by an IPlugin.

Parameters
layerNameName of the layer which the user wishes to validate.

Implements nvcaffeparser1::IPluginFactory.

Here is the call graph for this function:
Here is the caller graph for this function:

◆ isPluginExt()

bool PluginFactory::isPluginExt ( const char *  layerName)
inlineoverridevirtual

A user implemented function that determines if a layer configuration is provided by an IPluginExt.

Parameters
layerNameName of the layer which the user wishes to validate.

Implements nvcaffeparser1::IPluginFactoryExt.

Here is the caller graph for this function:

◆ createPlugin() [1/3]

virtual IPlugin* PluginFactory::createPlugin ( const char *  layerName,
const nvinfer1::Weights weights,
int  nbWeights 
)
inlineoverridevirtual
Here is the call graph for this function:

◆ createPlugin() [2/3]

nvinfer1::IPlugin* PluginFactory::createPlugin ( const char *  layerName,
const void *  serialData,
size_t  serialLength 
)
inlineoverridevirtual

Create a plugin from serialized data.

Responsibility of destroying this plugin lies with the application. It can be done anytime after consumers of this plugin are destroyed.

Parameters
layerNameThe name of the layer.
serialDataThe serialized data.
serialLengthThe length of the serialized data.
Returns
The plugin.
See also
IPlugin::serialize()

Implements nvinfer1::IPluginFactory.

Here is the call graph for this function:

◆ destroyPlugin()

void PluginFactory::destroyPlugin ( )
inline

◆ getVersion()

virtual int32_t nvcaffeparser1::IPluginFactoryExt::getVersion ( ) const
inlinevirtualinherited

◆ createPlugin() [3/3]

virtual nvinfer1::IPlugin* nvcaffeparser1::IPluginFactory::createPlugin ( const char *  layerName,
const nvinfer1::Weights weights,
int32_t  nbWeights 
)
pure virtualinherited

Creates a plugin.

Parameters
layerNameName of layer associated with the plugin.
weightsWeights used for the layer.
nbWeightsNumber of weights.

Member Data Documentation

◆ mPlugin

std::unique_ptr<FCPlugin> PluginFactory::mPlugin {nullptr}

The documentation for this class was generated from the following file: