TensorRT
7.1.3.0
|
A LRN layer in a network definition. More...
#include <NvInfer.h>
Public Member Functions | |
virtual void | setWindowSize (int windowSize)=0 |
Set the LRN window size. More... | |
virtual int | 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 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... | |
A LRN layer in a network definition.
The output size is the same as the input size.
|
pure virtual |
Get the LRN alpha value.
|
pure virtual |
Get the LRN beta value.
|
pure virtual |
Get the LRN K value.
|
pure virtual |
Get the LRN window size.
|
pure virtual |
|
pure virtual |
|
pure virtual |
|
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.