Struct IConvolutionLayer 

pub struct IConvolutionLayer { /* private fields */ }

! ! \class IConvolutionLayer ! ! \brief A convolution layer in a network definition. ! ! This layer performs a correlation operation between 3 or 4 dimensional filter with a 4 or 5 dimensional tensor to ! produce another 4 or 5 dimensional tensor. ! ! An optional bias argument is supported, which adds a per-channel constant to each value in the output. ! ! \warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI. !

Implementations

impl IConvolutionLayer

pub fn setNbOutputMaps(self: Pin<&mut IConvolutionLayer>, nbOutputMaps: i64)

! ! \brief Set the number of output maps for the convolution. ! ! If executing this layer on DLA, the number of output maps must be in the range [1,8192]. ! ! \see getNbOutputMaps() !

pub fn getNbOutputMaps(self: &IConvolutionLayer) -> i64

! ! \brief Get the number of output maps for the convolution. ! ! \see setNbOutputMaps() !

pub fn setNbGroups(self: Pin<&mut IConvolutionLayer>, nbGroups: i64)

! ! \brief Set the number of groups for a convolution. ! ! The input tensor channels are divided into \p nbGroups groups, and a convolution is executed for each group, ! using a filter per group. The results of the group convolutions are concatenated to form the output. ! ! \note When using groups in int8 mode, the size of the groups (i.e. the channel count divided by the group ! count) must be a multiple of 4 for both input and output. ! ! Default: 1 ! ! If executing this layer on DLA, the max number of groups is 8192. ! ! \see getNbGroups() !

pub fn getNbGroups(self: &IConvolutionLayer) -> i64

! ! \brief Get the number of groups of the convolution. ! ! \see setNbGroups() !

pub fn setKernelWeights(self: Pin<&mut IConvolutionLayer>, weights: Weights)

! ! \brief Set the kernel weights for the convolution. ! ! The weights are specified as a contiguous array in \p GKCRS order, where \p G is the number of groups, \p K ! the number of output feature maps, \p C the number of input channels, and \p R and \p S are the height and ! width of the filter. ! ! \see getKernelWeights() !

pub fn getKernelWeights(self: &IConvolutionLayer) -> Weights

! ! \brief Get the kernel weights of the convolution. ! ! \see setKernelWeights() !

pub fn setBiasWeights(self: Pin<&mut IConvolutionLayer>, weights: Weights)

! ! \brief Set the bias weights for the convolution. ! ! Bias is optional. To omit bias, set the count value of the weights structure to zero. ! ! The bias is applied per-channel, so the number of weights (if non-zero) must be equal to the number of output ! feature maps. ! ! \see getBiasWeights() !

pub fn getBiasWeights(self: &IConvolutionLayer) -> Weights

! ! \brief Get the bias weights for the convolution. ! ! \see setBiasWeights() !

pub fn setPrePadding(self: Pin<&mut IConvolutionLayer>, padding: &Dims64)

! ! \brief Set the multi-dimension pre-padding of the convolution. ! ! The start of the input will be zero-padded by this number of elements in each dimension. ! ! Default: (0, 0, …, 0) ! ! If executing this layer on DLA, only support 2D padding, both height and width of padding must be in the range ! [0,31], and the padding must be less than the kernel size. ! ! \see getPrePadding() !

pub fn getPrePadding(self: &IConvolutionLayer) -> Dims64

! ! \brief Get the pre-padding. ! ! \see setPrePadding() !

pub fn setPostPadding(self: Pin<&mut IConvolutionLayer>, padding: &Dims64)

! ! \brief Set the multi-dimension post-padding of the convolution. ! ! The end of the input will be zero-padded by this number of elements in each dimension. ! ! Default: (0, 0, …, 0) ! ! If executing this layer on DLA, only support 2D padding, both height and width of padding must be in the range ! [0,31], and the padding must be less than the kernel size. ! ! \see getPostPadding() !

pub fn getPostPadding(self: &IConvolutionLayer) -> Dims64

! ! \brief Get the post-padding. ! ! \see setPostPadding() !

pub fn setPaddingMode( self: Pin<&mut IConvolutionLayer>, paddingMode: PaddingMode, )

! ! \brief Set the padding mode. ! ! Padding mode takes precedence if both setPaddingMode and setPre/PostPadding are used. ! ! Default: kEXPLICIT_ROUND_DOWN ! ! \see getPaddingMode() !

pub fn getPaddingMode(self: &IConvolutionLayer) -> PaddingMode

! ! \brief Get the padding mode. ! ! Default: kEXPLICIT_ROUND_DOWN ! ! \see setPaddingMode() !

pub fn setKernelSizeNd(self: Pin<&mut IConvolutionLayer>, kernelSize: &Dims64)

! ! \brief Set the multi-dimension kernel size of the convolution. ! ! If executing this layer on DLA, only support 2D kernel size, both height and width of kernel size must be in the ! range [1,32]. ! ! \see getKernelSizeNd() !

pub fn getKernelSizeNd(self: &IConvolutionLayer) -> Dims64

! ! \brief Get the multi-dimension kernel size of the convolution. ! ! \see setKernelSizeNd() !

pub fn setStrideNd(self: Pin<&mut IConvolutionLayer>, stride: &Dims64)

! ! \brief Set the multi-dimension stride of the convolution. ! ! Default: (1, 1, …, 1) ! ! If executing this layer on DLA, only support 2D stride, both height and width of stride must be in the range ! [1,8]. ! ! \see getStrideNd() !

pub fn getStrideNd(self: &IConvolutionLayer) -> Dims64

! ! \brief Get the multi-dimension stride of the convolution. ! ! \see setStrideNd() !

pub fn setPaddingNd(self: Pin<&mut IConvolutionLayer>, padding: &Dims64)

! ! \brief Set the multi-dimension padding of the convolution. ! ! The input will be zero-padded by this number of elements in each dimension. ! Padding is symmetric. ! ! Default: (0, 0, …, 0) ! ! If executing this layer on DLA, only support 2D padding, both height and width of padding must be in the range ! [0,31], and the padding must be less than the kernel size. ! ! \see getPaddingNd() setPadding() getPadding() !

pub fn getPaddingNd(self: &IConvolutionLayer) -> Dims64

! ! \brief Get the multi-dimension padding of the convolution. ! ! If the padding is asymmetric, the pre-padding is returned. ! ! \see setPaddingNd() !

pub fn setDilationNd(self: Pin<&mut IConvolutionLayer>, dilation: &Dims64)

! ! \brief Set the multi-dimension dilation of the convolution. ! ! Default: (1, 1, …, 1) ! ! If executing this layer on DLA, only support 2D padding, both height and width must be in the range [1,32]. ! ! \see getDilationNd() !

pub fn getDilationNd(self: &IConvolutionLayer) -> Dims64

! ! \brief Get the multi-dimension dilation of the convolution. ! ! \see setDilationNd() !

Trait Implementations

impl AsRef<ILayer> for IConvolutionLayer

fn as_ref(self: &IConvolutionLayer) -> &ILayer

Converts this type into a shared reference of the (usually inferred) input type.

impl ExternType for IConvolutionLayer

type Id = (n, v, i, n, f, e, r, _1, (), I, C, o, n, v, o, l, u, t, i, o, n, L, a, y, e, r)

A type-level representation of the type’s C++ namespace and type name.

type Kind = Opaque

Either cxx::kind::Opaque or cxx::kind::Trivial.

impl MakeCppStorage for IConvolutionLayer

unsafe fn allocate_uninitialized_cpp_storage() -> *mut IConvolutionLayer

Allocates heap space for this type in C++ and return a pointer to that space, but do not initialize that space (i.e. do not yet call a constructor).

unsafe fn free_uninitialized_cpp_storage(arg0: *mut IConvolutionLayer)

Frees a C++ allocation which has not yet had a constructor called.