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

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IDeconvolutionLayer:

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

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

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

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

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

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

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

virtual DimsHW	getPadding () const =0
	Get the padding of the deconvolution. More...

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

virtual int	getNbGroups () const =0
	Set the number of groups for a deconvolution. More...

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

virtual Weights	getKernelWeights () const =0
	Get the kernel weights for the deconvolution. More...

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

virtual Weights	getBiasWeights () const =0
	Get the bias weights for the deconvolution. 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...

Detailed Description

A deconvolution layer in a network definition.

The output size is defined using the formula set by INetworkDefinition::setDeconvolutionOutputDimensionsFormula().

Member Function Documentation

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

pure virtual

Get the bias weights for the deconvolution.

See Also: getBiasWeights()

virtual DimsHW nvinfer1::IDeconvolutionLayer::getKernelSize ( ) const

pure virtual

Get the HW kernel size of the deconvolution.

See Also: setKernelSize()

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

pure virtual

Get the kernel weights for the deconvolution.

See Also: setNbGroups()

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

pure virtual

Set the number of groups for a deconvolution.

See Also: setNbGroups()

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

pure virtual

Get the number of output feature maps for the deconvolution.

See Also: setNbOutputMaps()

virtual DimsHW nvinfer1::IDeconvolutionLayer::getPadding ( ) const

pure virtual

Get the padding of the deconvolution.

See Also: setPadding()

virtual DimsHW nvinfer1::IDeconvolutionLayer::getStride ( ) const

pure virtual

Get the stride of the deconvolution.

Default: (1,1)

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

pure virtual

Set the bias weights for the deconvolution.

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

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

See Also: getBiasWeights()

virtual void nvinfer1::IDeconvolutionLayer::setKernelSize ( DimsHW kernelSize )

pure virtual

Set the HW kernel size of the convolution.

See Also: getKernelSize()

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

pure virtual

Set the kernel weights for the deconvolution.

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

See Also: getWeights()

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

pure virtual

Set the number of groups for a deconvolution.

The input tensor channels are divided into nbGroups groups, and a deconvolution 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