nvinfer1::IPaddingLayer Class Reference

Layer that represents a padding operation. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IPaddingLayer:

Public Member Functions
virtual void	setPrePadding (DimsHW padding)=0
	Set the padding that is applied at the start of the tensor. More...
virtual DimsHW	getPrePadding () const =0
	Get the padding that is applied at the start of the tensor. More...
virtual void	setPostPadding (DimsHW padding)=0
	Set the padding that is applied at the end of the tensor. More...
virtual DimsHW	getPostPadding () const =0
	Get the padding that is applied at the end of the 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	setInput (int index, ITensor &tensor)=0
	replace an input of this layer with a specific tensor 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

Layer that represents a padding operation.

The padding layer adds zero-padding at the start and end of the input tensor. It only supports padding along the two innermost dimensions. Applying negative padding results in cropping of the input.

Warning

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

Member Function Documentation

getPostPadding()

virtual DimsHW nvinfer1::IPaddingLayer::getPostPadding ( ) const

pure virtual

Get the padding that is applied at the end of the tensor.

See also

setPostPadding

getPrePadding()

virtual DimsHW nvinfer1::IPaddingLayer::getPrePadding ( ) const

pure virtual

Get the padding that is applied at the start of the tensor.

See also

setPrePadding

setPostPadding()

virtual void nvinfer1::IPaddingLayer::setPostPadding ( DimsHW  padding )

pure virtual

Set the padding that is applied at the end of the tensor.

Negative padding results in trimming the edge by the specified amount

See also

getPostPadding

setPrePadding()

virtual void nvinfer1::IPaddingLayer::setPrePadding ( DimsHW  padding )

pure virtual

Set the padding that is applied at the start of the tensor.

Negative padding results in trimming the edge by the specified amount

See also

getPrePadding

---

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

• NvInfer.h