TensorRT  6.0.0.6
nvinfer1::IPaddingLayer Class Referenceabstract

Layer that represents a padding operation. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IPaddingLayer:
nvinfer1::ILayer

Public Member Functions

virtual void setPrePadding (DimsHW padding) TRTNOEXCEPT=0
 Set the padding that is applied at the start of the tensor. More...
 
virtual DimsHW getPrePadding () const TRTNOEXCEPT=0
 Get the padding that is applied at the start of the tensor. More...
 
virtual void setPostPadding (DimsHW padding) TRTNOEXCEPT=0
 Set the padding that is applied at the end of the tensor. More...
 
virtual DimsHW getPostPadding () const TRTNOEXCEPT=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 TRTNOEXCEPT=0
 Return the type of a layer. More...
 
virtual void setName (const char *name) TRTNOEXCEPT=0
 Set the name of a layer. More...
 
virtual const char * getName () const TRTNOEXCEPT=0
 Return the name of a layer. More...
 
virtual int getNbInputs () const TRTNOEXCEPT=0
 Get the number of inputs of a layer.
 
virtual ITensorgetInput (int index) const TRTNOEXCEPT=0
 Get the layer input corresponding to the given index. More...
 
virtual int getNbOutputs () const TRTNOEXCEPT=0
 Get the number of outputs of a layer.
 
virtual ITensorgetOutput (int index) const TRTNOEXCEPT=0
 Get the layer output corresponding to the given index. More...
 
virtual void setInput (int index, ITensor &tensor) TRTNOEXCEPT=0
 replace an input of this layer with a specific tensor More...
 
virtual void setPrecision (DataType dataType) TRTNOEXCEPT=0
 Set the computational precision of this layer. More...
 
virtual DataType getPrecision () const TRTNOEXCEPT=0
 get the computational precision of this layer More...
 
virtual bool precisionIsSet () const TRTNOEXCEPT=0
 whether the computational precision has been set for this layer More...
 
virtual void resetPrecision () TRTNOEXCEPT=0
 reset the computational precision for this layer More...
 
virtual void setOutputType (int index, DataType dataType) TRTNOEXCEPT=0
 Set the output type of this layer. More...
 
virtual DataType getOutputType (int index) const TRTNOEXCEPT=0
 get the output type of this layer More...
 
virtual bool outputTypeIsSet (int index) const TRTNOEXCEPT=0
 whether the output type has been set for this layer More...
 
virtual void resetOutputType (int index) TRTNOEXCEPT=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

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

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

See also
setPostPadding
virtual DimsHW nvinfer1::IPaddingLayer::getPrePadding ( ) const
pure virtual

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

See also
setPrePadding
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
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: