A convolution layer in a network definition. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IConvolutionLayer:

Public Member Functions
virtual TRT_DEPRECATED void	setKernelSize (DimsHW kernelSize)=0
	Set the HW kernel size of the convolution. More...

virtual TRT_DEPRECATED DimsHW	getKernelSize () const =0
	Get the HW kernel size of the convolution. More...

virtual void	setNbOutputMaps (int nbOutputMaps)=0
	Set the number of output maps for the convolution. More...

virtual int	getNbOutputMaps () const =0
	Get the number of output maps for the convolution. More...

virtual TRT_DEPRECATED void	setStride (DimsHW stride)=0
	Get the stride of the convolution. More...

virtual TRT_DEPRECATED DimsHW	getStride () const =0
	Get the stride of the convolution. More...

virtual TRT_DEPRECATED void	setPadding (DimsHW padding)=0
	Set the padding of the convolution. More...

virtual TRT_DEPRECATED DimsHW	getPadding () const =0
	Get the padding of the convolution. If the padding is asymmetric, the pre-padding is returned. More...

virtual void	setNbGroups (int nbGroups)=0
	Set the number of groups for a convolution. More...

virtual int	getNbGroups () const =0
	Get the number of groups of the convolution. More...

virtual void	setKernelWeights (Weights weights)=0
	Set the kernel weights for the convolution. More...

virtual Weights	getKernelWeights () const =0
	Get the kernel weights of the convolution. More...

virtual void	setBiasWeights (Weights weights)=0
	Set the bias weights for the convolution. More...

virtual Weights	getBiasWeights () const =0
	Get the bias weights for the convolution. More...

virtual TRT_DEPRECATED void	setDilation (DimsHW dilation)=0
	Set the dilation for a convolution. More...

virtual TRT_DEPRECATED DimsHW	getDilation () const =0
	Get the dilation for a convolution. More...

virtual void	setPrePadding (Dims padding)=0
	Set the pre-padding. More...

virtual Dims	getPrePadding () const =0
	Get the pre-padding. More...

virtual void	setPostPadding (Dims padding)=0
	Set the post-padding. More...

virtual Dims	getPostPadding () const =0
	Get the post-padding. More...

virtual void	setPaddingMode (PaddingMode paddingMode)=0
	Set the padding mode. More...

virtual PaddingMode	getPaddingMode () const =0
	Get the padding mode. More...

virtual void	setKernelSizeNd (Dims kernelSize)=0
	Set the multi-dimension kernel size of the convolution. More...

virtual Dims	getKernelSizeNd () const =0
	Get the multi-dimension kernel size of the convolution. More...

virtual void	setStrideNd (Dims stride)=0
	Set the multi-dimension stride of the convolution. More...

virtual Dims	getStrideNd () const =0
	Get the multi-dimension stride of the convolution. More...

virtual void	setPaddingNd (Dims padding)=0
	Set the multi-dimension padding of the convolution. More...

virtual Dims	getPaddingNd () const =0
	Get the multi-dimension padding of the convolution. More...

virtual void	setDilationNd (Dims dilation)=0
	Set the multi-dimension dilation of the convolution. More...

virtual Dims	getDilationNd () const =0
	Get the multi-dimension dilation of the convolution. More...

void	setInput (int index, ITensor &tensor) _TENSORRT_OVERRIDE=0
	Append or replace an input of this layer with a specific tensor. More...

Public Member Functions inherited from nvinfer1::ILayer
virtual LayerType	getType () const =0
	Return the type of a layer. More...

virtual void	setName (const char *name)=0
	Set the name of a layer. More...

virtual const char *	getName () const =0
	Return the name of a layer. More...

virtual int	getNbInputs () const =0
	Get the number of inputs of a layer.

virtual ITensor *	getInput (int index) const =0
	Get the layer input corresponding to the given index. More...

virtual int	getNbOutputs () const =0
	Get the number of outputs of a layer.

virtual ITensor *	getOutput (int index) const =0
	Get the layer output corresponding to the given index. More...

virtual void	setPrecision (DataType dataType)=0
	Set the computational precision of this layer. More...

virtual DataType	getPrecision () const =0
	get the computational precision of this layer More...

virtual bool	precisionIsSet () const =0
	whether the computational precision has been set for this layer More...

virtual void	resetPrecision ()=0
	reset the computational precision for this layer More...

virtual void	setOutputType (int index, DataType dataType)=0
	Set the output type of this layer. More...

virtual DataType	getOutputType (int index) const =0
	get the output type of this layer More...

virtual bool	outputTypeIsSet (int index) const =0
	whether the output type has been set for this layer More...

virtual void	resetOutputType (int index)=0
	reset the output type for this layer More...

Detailed Description

A convolution layer in a network definition.

This layer performs a correlation operation between 3-dimensional filter with a 4-dimensional tensor to produce another 4-dimensional tensor.

An optional bias argument is supported, which adds a per-channel constant to each value in the output.

Warning: Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Member Function Documentation

virtual Weights nvinfer1::IConvolutionLayer::getBiasWeights ( ) const

pure virtual

Get the bias weights for the convolution.

See also: setBiasWeights()

virtual TRT_DEPRECATED DimsHW nvinfer1::IConvolutionLayer::getDilation ( ) const

pure virtual

Get the dilation for a convolution.

See also: setDilation()

Deprecated:: Superseded by getDilationNd

virtual Dims nvinfer1::IConvolutionLayer::getDilationNd ( ) const

pure virtual

Get the multi-dimension dilation of the convolution.

See also: setDilation()

virtual TRT_DEPRECATED DimsHW nvinfer1::IConvolutionLayer::getKernelSize ( ) const

pure virtual

Get the HW kernel size of the convolution.

See also: setKernelSize()

Deprecated:: Superseded by getKernelSizeNd.

virtual Dims nvinfer1::IConvolutionLayer::getKernelSizeNd ( ) const

pure virtual

Get the multi-dimension kernel size of the convolution.

See also: setKernelSizeNd()

virtual Weights nvinfer1::IConvolutionLayer::getKernelWeights ( ) const

pure virtual

Get the kernel weights of the convolution.

See also: setKernelWeights()

virtual int nvinfer1::IConvolutionLayer::getNbGroups ( ) const

pure virtual

Get the number of groups of the convolution.

See also: setNbGroups()

virtual int nvinfer1::IConvolutionLayer::getNbOutputMaps ( ) const

pure virtual

Get the number of output maps for the convolution.

See also: setNbOutputMaps()

virtual TRT_DEPRECATED DimsHW nvinfer1::IConvolutionLayer::getPadding ( ) const

pure virtual

Get the padding of the convolution. If the padding is asymmetric, the pre-padding is returned.

See also: setPadding()

Deprecated:: Superseded by getPaddingNd

virtual PaddingMode nvinfer1::IConvolutionLayer::getPaddingMode ( ) const

pure virtual

Get the padding mode.

Default: kEXPLICIT_ROUND_DOWN

See also: setPaddingMode()

virtual Dims nvinfer1::IConvolutionLayer::getPaddingNd ( ) const

pure virtual

Get the multi-dimension padding of the convolution.

If the padding is asymmetric, the pre-padding is returned.

See also: setPaddingNd()

virtual Dims nvinfer1::IConvolutionLayer::getPostPadding ( ) const

pure virtual

Get the post-padding.

See also: setPostPadding()

virtual Dims nvinfer1::IConvolutionLayer::getPrePadding ( ) const

pure virtual

Get the pre-padding.

See also: setPrePadding()

virtual TRT_DEPRECATED DimsHW nvinfer1::IConvolutionLayer::getStride ( ) const

pure virtual

Get the stride of the convolution.

Deprecated:: Superseded by getStrideNd.

virtual Dims nvinfer1::IConvolutionLayer::getStrideNd ( ) const

pure virtual

Get the multi-dimension stride of the convolution.

See also: setStrideNd()

virtual void nvinfer1::IConvolutionLayer::setBiasWeights ( Weights weights )

pure virtual

Set the bias weights for the convolution.

Bias is optional. To omit bias, set the count value of the weights structure to zero.

The bias is applied per-channel, so the number of weights (if non-zero) must be equal to the number of output feature maps.

See also: getBiasWeights()

virtual TRT_DEPRECATED void nvinfer1::IConvolutionLayer::setDilation ( DimsHW dilation )

pure virtual

Set the dilation for a convolution.

Default: (1,1)

See also: getDilation()

Deprecated:: Superseded by setDilationNd

virtual void nvinfer1::IConvolutionLayer::setDilationNd ( Dims dilation )

pure virtual

Set the multi-dimension dilation of the convolution.

Default: (1, 1, ..., 1)

See also: getDilation()

void nvinfer1::IConvolutionLayer::setInput	(	int	index,
		ITensor &	tensor
	)

pure virtual

Append or replace an input of this layer with a specific tensor.

Parameters

index	the index of the input to modify.
tensor	the new input tensor

For a convolution layer, the values 0-2 are valid. The value 1 override kernel weights, and the value 2 override bias weights. Conversely, this input tensor can be overridden via appropriate set call. The indices are as follows:

Index | Description 0 | The input activation tensor. 1 | The kernel weights tensor. 2 | The bias weights tensor.

If this function is called with a value greater than 0, then the function getNbInputs() changes

Implements nvinfer1::ILayer.

virtual TRT_DEPRECATED void nvinfer1::IConvolutionLayer::setKernelSize ( DimsHW kernelSize )

pure virtual

Set the HW kernel size of the convolution.

If executing this layer on DLA, both height and width of kernel size must be in the range [1,16].

See also: getKernelSize()

Deprecated:: Superseded by setKernelSizeNd.

virtual void nvinfer1::IConvolutionLayer::setKernelSizeNd ( Dims kernelSize )

pure virtual

Set the multi-dimension kernel size of the convolution.

If executing this layer on DLA, only support 2D kernel size, both height and width of kernel size must be in the range [1,16].

See also: getKernelSizeNd()

virtual void nvinfer1::IConvolutionLayer::setKernelWeights ( Weights weights )

pure virtual

Set the kernel weights for the convolution.

The weights are specified as a contiguous array in GKCRS order, where G is the number of groups, K the number of output feature maps, C the number of input channels, and R and S are the height and width of the filter.

See also: getKernelWeights()

virtual void nvinfer1::IConvolutionLayer::setNbGroups ( int nbGroups )

pure virtual

Set the number of groups for a convolution.

The input tensor channels are divided into nbGroups groups, and a convolution is executed for each group, using a filter per group. The results of the group convolutions are concatenated to form the output.

Note: When using groups in int8 mode, the size of the groups (i.e. the channel count divided by the group count) must be a multiple of 4 for both input and output.

Default: 1

See also: getNbGroups()

virtual void nvinfer1::IConvolutionLayer::setNbOutputMaps ( int nbOutputMaps )

pure virtual

Set the number of output maps for the convolution.

If executing this layer on DLA, the number of output maps must be in the range [1,8192].

See also: getNbOutputMaps()

virtual TRT_DEPRECATED void nvinfer1::IConvolutionLayer::setPadding ( DimsHW padding )

pure virtual

Set the padding of the convolution.

The input will be zero-padded by this number of elements in the height and width directions. Padding is symmetric.

Default: (0,0)

If executing this layer on DLA, both height and width of padding must be in the range [0,15].

See also: getPadding()

Deprecated:: Superseded by setPaddingNd

virtual void nvinfer1::IConvolutionLayer::setPaddingMode ( PaddingMode paddingMode )

pure virtual

Set the padding mode.

Padding mode takes precedence if both setPaddingMode and setPre/PostPadding are used.

Default: kEXPLICIT_ROUND_DOWN

See also: getPaddingMode()

virtual void nvinfer1::IConvolutionLayer::setPaddingNd ( Dims padding )

pure virtual

Set the multi-dimension padding of the convolution.

The input will be zero-padded by this number of elements in each dimension. Padding is symmetric.

Default: (0, 0, ..., 0)

If executing this layer on DLA, only support 2D padding, both height and width of padding must be in the range [0,15].

See also: getPaddingNd() setPadding() getPadding()

virtual void nvinfer1::IConvolutionLayer::setPostPadding ( Dims padding )

pure virtual

Set the post-padding.

The end of the input will be zero-padded by this number of elements in the height and width directions.

Default: (0,0)

If executing this layer on DLA, both height and width of padding must be in the range [0,15].

See also: getPostPadding()

virtual void nvinfer1::IConvolutionLayer::setPrePadding ( Dims padding )

pure virtual

Set the pre-padding.

The start of input will be zero-padded by this number of elements in the height and width directions.

Default: 0

If executing this layer on DLA, both height and width of padding must be in the range [0,15].

See also: getPrePadding()

virtual TRT_DEPRECATED void nvinfer1::IConvolutionLayer::setStride ( DimsHW stride )

pure virtual

Get the stride of the convolution.

Default: (1,1)

If executing this layer on DLA, both height and width of stride must be in the range [1,8].

See also: getStride()

Deprecated:: Superseded by setStrideNd.

virtual void nvinfer1::IConvolutionLayer::setStrideNd ( Dims stride )

pure virtual

Set the multi-dimension stride of the convolution.

Default: (1, 1, ..., 1)

If executing this layer on DLA, only support 2D stride, both height and width of stride must be in the range [1,8].

See also: getStrideNd() setStride() getStride()

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

NvInfer.h

Public Member Functions

Detailed Description

Member Function Documentation