Plugins

class tensorrt.ICaffePluginFactoryV2

Plugin factory used to configure plugins.

create_plugin(self: tensorrt.tensorrt.ICaffePluginFactoryV2, layer_name: str, weights: std::vector<nvinfer1::Weights, std::allocator<nvinfer1::Weights> >) → tensorrt.tensorrt.IPluginV2

Creates a plugin.

arg layer_name

Name of layer associated with the plugin.

arg weights

Weights used for the layer.

Returns

The newly created IPluginV2 .

is_plugin_v2(self: tensorrt.tensorrt.ICaffePluginFactoryV2, layer_name: str) → bool

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

Parameters

layer_name – Name of the layer which the user wishes to validate.

Returns

True if the the layer configuration is provided by an IPluginV2 .