nvinfer1::ILRNLayer Class Reference

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

#include <NvInfer.h>

Inheritance diagram for nvinfer1::ILRNLayer:

Public Member Functions
virtual void	setWindowSize (int32_t windowSize)=0
	Set the LRN window size. More...
virtual int32_t	getWindowSize () const =0
	Get the LRN window size. More...
virtual void	setAlpha (float alpha)=0
	Set the LRN alpha value. More...
virtual float	getAlpha () const =0
	Get the LRN alpha value. More...
virtual void	setBeta (float beta)=0
	Set the LRN beta value. More...
virtual float	getBeta () const =0
	Get the LRN beta value. More...
virtual void	setK (float k)=0
	Set the LRN K value. More...
virtual float	getK () const =0
	Get the LRN K value. 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 int32_t	getNbInputs () const =0
	Get the number of inputs of a layer.
virtual ITensor *	getInput (int32_t index) const =0
	Get the layer input corresponding to the given index. More...
virtual int32_t	getNbOutputs () const =0
	Get the number of outputs of a layer.
virtual ITensor *	getOutput (int32_t index) const =0
	Get the layer output corresponding to the given index. More...
virtual void	setInput (int32_t 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 (int32_t index, DataType dataType)=0
	Set the output type of this layer. More...
virtual DataType	getOutputType (int32_t index) const =0
	get the output type of this layer More...
virtual bool	outputTypeIsSet (int32_t index) const =0
	whether the output type has been set for this layer More...
virtual void	resetOutputType (int32_t index)=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

getAlpha()

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

pure virtual

Get the LRN alpha value.

See also

setAlpha()

getBeta()

virtual float nvinfer1::ILRNLayer::getBeta ( ) const

pure virtual

Get the LRN beta value.

See also

setBeta()

getK()

virtual float nvinfer1::ILRNLayer::getK ( ) const

pure virtual

Get the LRN K value.

See also

setK()

getWindowSize()

virtual int32_t nvinfer1::ILRNLayer::getWindowSize ( ) const

pure virtual

Get the LRN window size.

See also

getWindowStride()

setAlpha()

virtual void nvinfer1::ILRNLayer::setAlpha ( float  alpha )

pure virtual

Set the LRN alpha value.

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

See also

getAlpha()

setBeta()

virtual void nvinfer1::ILRNLayer::setBeta ( float  beta )

pure virtual

Set the LRN beta value.

The valid range is [0.01, 1e5f].

See also

getBeta()

setK()

virtual void nvinfer1::ILRNLayer::setK ( float  k )

pure virtual

Set the LRN K value.

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

See also

getK()

setWindowSize()

virtual void nvinfer1::ILRNLayer::setWindowSize ( int32_t  windowSize )

pure virtual

Set the LRN window size.

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

If executing this layer on the DLA, only values in the set, [3, 5, 7, 9], are valid.

See also

setWindowStride()

---

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

• NvInfer.h