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 .