TensorRT 8.2.5
|
A deconvolution layer in a network definition. More...
#include <NvInfer.h>
Public Member Functions | |
TRT_DEPRECATED void | setKernelSize (DimsHW kernelSize) noexcept |
Set the HW kernel size of the convolution. More... | |
TRT_DEPRECATED DimsHW | getKernelSize () const noexcept |
Get the HW kernel size of the deconvolution. More... | |
void | setNbOutputMaps (int32_t nbOutputMaps) noexcept |
Set the number of output feature maps for the deconvolution. More... | |
int32_t | getNbOutputMaps () const noexcept |
Get the number of output feature maps for the deconvolution. More... | |
TRT_DEPRECATED void | setStride (DimsHW stride) noexcept |
Set the stride of the deconvolution. More... | |
TRT_DEPRECATED DimsHW | getStride () const noexcept |
Get the stride of the deconvolution. More... | |
TRT_DEPRECATED void | setPadding (DimsHW padding) noexcept |
Set the padding of the deconvolution. More... | |
TRT_DEPRECATED DimsHW | getPadding () const noexcept |
Get the padding of the deconvolution. More... | |
void | setNbGroups (int32_t nbGroups) noexcept |
Set the number of groups for a deconvolution. More... | |
int32_t | getNbGroups () const noexcept |
Get the number of groups for a deconvolution. More... | |
void | setKernelWeights (Weights weights) noexcept |
Set the kernel weights for the deconvolution. More... | |
Weights | getKernelWeights () const noexcept |
Get the kernel weights for the deconvolution. More... | |
void | setBiasWeights (Weights weights) noexcept |
Set the bias weights for the deconvolution. More... | |
Weights | getBiasWeights () const noexcept |
Get the bias weights for the deconvolution. More... | |
void | setPrePadding (Dims padding) noexcept |
Set the multi-dimension pre-padding of the deconvolution. More... | |
Dims | getPrePadding () const noexcept |
Get the pre-padding. More... | |
void | setPostPadding (Dims padding) noexcept |
Set the multi-dimension post-padding of the deconvolution. More... | |
Dims | getPostPadding () const noexcept |
Get the padding. More... | |
void | setPaddingMode (PaddingMode paddingMode) noexcept |
Set the padding mode. More... | |
PaddingMode | getPaddingMode () const noexcept |
Get the padding mode. More... | |
void | setKernelSizeNd (Dims kernelSize) noexcept |
Set the multi-dimension kernel size of the deconvolution. More... | |
Dims | getKernelSizeNd () const noexcept |
Get the multi-dimension kernel size of the deconvolution. More... | |
void | setStrideNd (Dims stride) noexcept |
Set the multi-dimension stride of the deconvolution. More... | |
Dims | getStrideNd () const noexcept |
Get the multi-dimension stride of the deconvolution. More... | |
void | setPaddingNd (Dims padding) noexcept |
Set the multi-dimension padding of the deconvolution. More... | |
Dims | getPaddingNd () const noexcept |
Get the multi-dimension padding of the deconvolution. More... | |
void | setDilationNd (Dims dilation) noexcept |
Set the multi-dimension dilation of the deconvolution. More... | |
Dims | getDilationNd () const noexcept |
Get the multi-dimension dilation of the deconvolution. More... | |
void | setInput (int32_t index, ITensor &tensor) noexcept |
Append or replace an input of this layer with a specific tensor. More... | |
![]() | |
LayerType | getType () const noexcept |
Return the type of a layer. More... | |
void | setName (const char *name) noexcept |
Set the name of a layer. More... | |
const char * | getName () const noexcept |
Return the name of a layer. More... | |
int32_t | getNbInputs () const noexcept |
Get the number of inputs of a layer. | |
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. | |
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 computational precision of this layer. 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. 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... | |
Protected Attributes | |
apiv::VDeconvolutionLayer * | mImpl |
![]() | |
apiv::VLayer * | mLayer |
Additional Inherited Members | |
![]() | |
INoCopy (const INoCopy &other)=delete | |
INoCopy & | operator= (const INoCopy &other)=delete |
INoCopy (INoCopy &&other)=delete | |
INoCopy & | operator= (INoCopy &&other)=delete |
A deconvolution layer in a network definition.
|
inlinenoexcept |
Get the bias weights for the deconvolution.
|
inlinenoexcept |
Get the multi-dimension dilation of the deconvolution.
|
inlinenoexcept |
Get the HW kernel size of the deconvolution.
|
inlinenoexcept |
Get the multi-dimension kernel size of the deconvolution.
|
inlinenoexcept |
Get the kernel weights for the deconvolution.
|
inlinenoexcept |
Get the number of groups for a deconvolution.
|
inlinenoexcept |
Get the number of output feature maps for the deconvolution.
|
inlinenoexcept |
Get the padding of the deconvolution.
Default: (0, 0)
|
inlinenoexcept |
|
inlinenoexcept |
Get the multi-dimension padding of the deconvolution.
If the padding is asymmetric, the pre-padding is returned.
|
inlinenoexcept |
Get the padding.
|
inlinenoexcept |
Get the pre-padding.
|
inlinenoexcept |
Get the stride of the deconvolution.
Default: (1,1)
|
inlinenoexcept |
Get the multi-dimension stride of the deconvolution.
|
inlinenoexcept |
Set the bias weights for the deconvolution.
Bias is optional. To omit bias, set the count value of the weights structure to zero.
The bias is applied per-feature-map, so the number of weights (if non-zero) must be equal to the number of output feature maps.
|
inlinenoexcept |
Set the multi-dimension dilation of the deconvolution.
Default: (1, 1, ..., 1)
|
inlinenoexcept |
Append or replace an input of this layer with a specific tensor.
index | the index of the input to modify. |
tensor | the new input tensor |
Only index 0 (data input) is valid, unless explicit-quantization mode is enabled. In explicit-quantization mode, input with index 1 is the kernel-weights tensor, if present. The kernel-weights tensor must be a build-time constant (computable at build-time via constant-folding) and an output of a dequantize layer. If input index 1 is used then the kernel-weights parameter must be set to empty Weights.
The indices are as follows:
If this function is called with the value 1, then the function getNbInputs() changes from returning 1 to 2.
|
inlinenoexcept |
Set the HW kernel size of the convolution.
If executing this layer on DLA, both height and width of kernel size must be in the range [1,32], or the combinations of [64, 96, 128] in one dimension and 1 in the other dimensions, i.e. [1x64] or [64x1] are valid, but not [64x64].
|
inlinenoexcept |
Set the multi-dimension kernel size of the deconvolution.
If executing this layer on DLA, there are ttwo restrictions: 1) Only 2D Kernel is supported. 2) Kernel height and width must be in the range [1,32] or the combinations of [64, 96, 128] in one dimension and 1 in the other dimensions, i.e. [1x64] or [64x1] are valid, but not [64x64].
|
inlinenoexcept |
Set the kernel weights for the deconvolution.
The weights are specified as a contiguous array in CKRS
order, where C
the number of input channels, K
the number of output feature maps, and R
and S
are the height and width of the filter.
|
inlinenoexcept |
Set the number of groups for a deconvolution.
The input tensor channels are divided into nbGroups
groups, and a deconvolution is executed for each group, using a filter per group. The results of the group convolutions are concatenated to form the output.
If executing this layer on DLA, nbGroups must be one
Default: 1
|
inlinenoexcept |
Set the number of output feature maps for the deconvolution.
If executing this layer on DLA, the number of output maps must be in the range [1,8192].
|
inlinenoexcept |
Set the padding of the deconvolution.
The output will be trimmed by this number of elements on each side in the height and width directions. In other words, it resembles the inverse of a convolution layer with this padding size. Padding is symmetric, and negative padding is not supported.
Default: (0,0)
If executing this layer on DLA, both height and width of padding must be 0.
|
inlinenoexcept |
Set the padding mode.
Padding mode takes precedence if both setPaddingMode and setPre/PostPadding are used.
Default: kEXPLICIT_ROUND_DOWN
|
inlinenoexcept |
Set the multi-dimension padding of the deconvolution.
The output will be trimmed by this number of elements on both sides of every dimension. In other words, it resembles the inverse of a convolution layer with this padding size. Padding is symmetric, and negative padding is not supported.
Default: (0, 0, ..., 0)
If executing this layer on DLA, padding must be 0.
|
inlinenoexcept |
Set the multi-dimension post-padding of the deconvolution.
The output will be trimmed by this number of elements on the end of every dimension. In other words, it resembles the inverse of a convolution layer with this padding size. Negative padding is not supported.
Default: (0, 0, ..., 0)
If executing this layer on DLA, padding must be 0.
|
inlinenoexcept |
Set the multi-dimension pre-padding of the deconvolution.
The output will be trimmed by this number of elements on the start of every dimension. In other words, it resembles the inverse of a convolution layer with this padding size. Negative padding is not supported.
Default: (0, 0, ..., 0)
If executing this layer on DLA, padding must be 0.
|
inlinenoexcept |
Set the stride of the deconvolution.
If executing this layer on DLA, there are two restrictions: 1) Stride height and width must be in the range [1,32] or the combinations of [64, 96, 128] in one dimension and 1 in the other dimensions, i.e. [1x64] or [64x1] are valid, but not [64x64]. 2) Stride values in each dimension must be equal to the corresponding kernel dimension values.
|
inlinenoexcept |
Set the multi-dimension stride of the deconvolution.
Default: (1, 1, ..., 1)
If executing this layer on DLA, there are three restrictions: 1) Only 2D Stride is supported. 2) Stride height and width must be in the range [1,32] or the combinations of [64, 96, 128] in one dimension and 1 in the other dimensions, i.e. [1x64] or [64x1] are valid, but not [64x64]. 3) Stride values in each dimension must be equal to the corresponding kernel dimension values.
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