Plugins¶
-
class
tensorrt.
IUffPluginFactory
(self: tensorrt.tensorrt.IUffPluginFactory) → None¶ Plugin factory used to configure plugins.
-
create_plugin
(self: tensorrt.tensorrt.IUffPluginFactory, layer_name: str, weights: std::vector<nvinfer1::Weights, std::allocator<nvinfer1::Weights> >, field_collection: tensorrt.tensorrt.FieldCollection) → tensorrt.tensorrt.IPlugin¶ Creates a plugin.
- arg layer_name
Name of layer associated with the plugin.
- arg weights
Weights used for the layer.
- arg field_collection
A collection of FieldMaps used as layer parameters for different plugin layers.
- Returns
The newly created
IPlugin
.
-
is_plugin
(self: tensorrt.tensorrt.IUffPluginFactory, layer_name: str) → bool¶ A user implemented function that determines if a layer configuration is provided by an
IPlugin
.- Parameters
layer_name – Name of the layer which the user wishes to validate.
- Returns
True if the the layer configuration is provided by an
IPlugin
.
-
-
class
tensorrt.
IUffPluginFactoryExt
¶ Plugin factory used to configure plugins with added support for TRT versioning.
-
get_version
(self: tensorrt.tensorrt.IUffPluginFactoryExt) → int¶ Get the Tensorrt Version
-
is_plugin_ext
(self: tensorrt.tensorrt.IUffPluginFactoryExt, layer_name: str) → bool¶ A user implemented function that determines if a layer configuration is provided by an
IPluginExt
.- Parameters
layer_name – Name of the layer which the user wishes to validate.
- Returns
True if the the layer configuration is provided by an
IPluginExt
.
-