TensorRT  6.0.0.6
nvinfer1::ILRNLayer Class Referenceabstract

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

#include <NvInfer.h>

Inheritance diagram for nvinfer1::ILRNLayer:
nvinfer1::ILayer

Public Member Functions

virtual void setWindowSize (int windowSize) TRTNOEXCEPT=0
 Set the LRN window size. More...
 
virtual int getWindowSize () const TRTNOEXCEPT=0
 Get the LRN window size. More...
 
virtual void setAlpha (float alpha) TRTNOEXCEPT=0
 Set the LRN alpha value. More...
 
virtual float getAlpha () const TRTNOEXCEPT=0
 Get the LRN alpha value. More...
 
virtual void setBeta (float beta) TRTNOEXCEPT=0
 Set the LRN beta value. More...
 
virtual float getBeta () const TRTNOEXCEPT=0
 Get the LRN beta value. More...
 
virtual void setK (float k) TRTNOEXCEPT=0
 Set the LRN K value. More...
 
virtual float getK () const TRTNOEXCEPT=0
 Get the LRN K value. 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

A LRN layer in a network definition.

The output size is the same as the input size.

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

Member Function Documentation

virtual float nvinfer1::ILRNLayer::getAlpha ( ) const
pure virtual

Get the LRN alpha value.

See also
setAlpha()
virtual float nvinfer1::ILRNLayer::getBeta ( ) const
pure virtual

Get the LRN beta value.

See also
setBeta()
virtual float nvinfer1::ILRNLayer::getK ( ) const
pure virtual

Get the LRN K value.

See also
setK()
virtual int nvinfer1::ILRNLayer::getWindowSize ( ) const
pure virtual

Get the LRN window size.

See also
getWindowStride()
virtual void nvinfer1::ILRNLayer::setAlpha ( float  alpha)
pure virtual

Set the LRN alpha value.

The valid range is [-1e20, 1e20].

See also
getAlpha()
virtual void nvinfer1::ILRNLayer::setBeta ( float  beta)
pure virtual

Set the LRN beta value.

The valid range is [0.01, 1e5f].

See also
getBeta()
virtual void nvinfer1::ILRNLayer::setK ( float  k)
pure virtual

Set the LRN K value.

The valid range is [1e-5, 1e10].

See also
getK()
virtual void nvinfer1::ILRNLayer::setWindowSize ( int  windowSize)
pure virtual

Set the LRN window size.

The window size must be odd and in the range of [1, 15].

See also
setWindowStride()

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