TensorRT 10.3.0
|
A LRN layer in a network definition. More...
#include <NvInfer.h>
Public Member Functions | |
void | setWindowSize (int64_t windowSize) noexcept |
Set the LRN window size. More... | |
int64_t | getWindowSize () const noexcept |
Get the LRN window size. More... | |
void | setAlpha (float alpha) noexcept |
Set the LRN alpha value. More... | |
float | getAlpha () const noexcept |
Get the LRN alpha value. More... | |
void | setBeta (float beta) noexcept |
Set the LRN beta value. More... | |
float | getBeta () const noexcept |
Get the LRN beta value. More... | |
void | setK (float k) noexcept |
Set the LRN K value. More... | |
float | getK () const noexcept |
Get the LRN K value. More... | |
![]() | |
LayerType | getType () const noexcept |
Return the type of a layer. More... | |
void | setName (char const *name) noexcept |
Set the name of a layer. More... | |
char const * | getName () const noexcept |
Return the name of a layer. More... | |
int32_t | getNbInputs () const noexcept |
Get the number of inputs of a layer. More... | |
ITensor * | getInput (int32_t index) const noexcept |
Get the layer input corresponding to the given index. More... | |
int32_t | getNbOutputs () const noexcept |
Get the number of outputs of a layer. More... | |
ITensor * | getOutput (int32_t index) const noexcept |
Get the layer output corresponding to the given index. More... | |
void | setInput (int32_t index, ITensor &tensor) noexcept |
Replace an input of this layer with a specific tensor. More... | |
void | setPrecision (DataType dataType) noexcept |
Set the preferred or required computational precision of this layer in a weakly-typed network. More... | |
DataType | getPrecision () const noexcept |
get the computational precision of this layer More... | |
bool | precisionIsSet () const noexcept |
whether the computational precision has been set for this layer More... | |
void | resetPrecision () noexcept |
reset the computational precision for this layer More... | |
void | setOutputType (int32_t index, DataType dataType) noexcept |
Set the output type of this layer in a weakly-typed network. More... | |
DataType | getOutputType (int32_t index) const noexcept |
get the output type of this layer More... | |
bool | outputTypeIsSet (int32_t index) const noexcept |
whether the output type has been set for this layer More... | |
void | resetOutputType (int32_t index) noexcept |
reset the output type for this layer More... | |
void | setMetadata (char const *metadata) noexcept |
Set the metadata for this layer. More... | |
char const * | getMetadata () const noexcept |
Get the metadata of the layer. More... | |
Protected Member Functions | |
virtual | ~ILRNLayer () noexcept=default |
![]() | |
virtual | ~ILayer () noexcept=default |
![]() | |
INoCopy ()=default | |
virtual | ~INoCopy ()=default |
INoCopy (INoCopy const &other)=delete | |
INoCopy & | operator= (INoCopy const &other)=delete |
INoCopy (INoCopy &&other)=delete | |
INoCopy & | operator= (INoCopy &&other)=delete |
Protected Attributes | |
apiv::VLRNLayer * | mImpl |
![]() | |
apiv::VLayer * | mLayer |
A LRN layer in a network definition.
The output size is the same as the input size.
|
protectedvirtualdefaultnoexcept |
|
inlinenoexcept |
Get the LRN alpha value.
|
inlinenoexcept |
Get the LRN beta value.
|
inlinenoexcept |
Get the LRN K value.
|
inlinenoexcept |
Get the LRN window size.
|
inlinenoexcept |
|
inlinenoexcept |
|
inlinenoexcept |
|
inlinenoexcept |
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.
|
protected |
Copyright © 2024 NVIDIA Corporation
Privacy Policy |
Manage My Privacy |
Do Not Sell or Share My Data |
Terms of Service |
Accessibility |
Corporate Policies |
Product Security |
Contact