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 .