|
TensorRT 10.13.0
|
Layer that represents a constant value. More...
#include <NvInfer.h>
Public Member Functions | |
| void | setWeights (Weights weights) noexcept |
| Set the weights for the layer. More... | |
| Weights | getWeights () const noexcept |
| Get the weights for the layer. More... | |
| void | setDimensions (Dims const &dimensions) noexcept |
| Set the dimensions for the layer. More... | |
| Dims | getDimensions () const noexcept |
| Get the dimensions for the layer. More... | |
Public Member Functions inherited from nvinfer1::ILayer | |
| 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... | |
| TRT_DEPRECATED 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... | |
| TRT_DEPRECATED bool | precisionIsSet () const noexcept |
| whether the computational precision has been set for this layer More... | |
| TRT_DEPRECATED void | resetPrecision () noexcept |
| reset the computational precision for this layer More... | |
| TRT_DEPRECATED 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... | |
| TRT_DEPRECATED bool | outputTypeIsSet (int32_t index) const noexcept |
| whether the output type has been set for this layer More... | |
| TRT_DEPRECATED 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 | ~IConstantLayer () noexcept=default |
Protected Member Functions inherited from nvinfer1::ILayer | |
| virtual | ~ILayer () noexcept=default |
Protected Member Functions inherited from nvinfer1::INoCopy | |
| 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::VConstantLayer * | mImpl |
Protected Attributes inherited from nvinfer1::ILayer | |
| apiv::VLayer * | mLayer |
Layer that represents a constant value.
|
protectedvirtualdefaultnoexcept |
|
inlinenoexcept |
|
inlinenoexcept |
Get the weights for the layer.
|
inlinenoexcept |
Set the dimensions for the layer.
| dimensions | The dimensions of the layer |
|
inlinenoexcept |
Set the weights for the layer.
The output type is weights.type. If the network is weakly typed and the weights have a real type, the output type might be different per TensorRT's type conversion rules.
|
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