Show / Hide Table of Contents

    Enum cudnnBatchNormMode

    cudnnBatchNormMode is an enumerated type used to specify the mode of operation in cudnnBatchNormalizationForwardInference(), cudnnBatchNormalizationForwardTraining(), cudnnBatchNormalizationBackward() and cudnnDeriveBNTensorDescriptor() routines.

    Namespace: ManagedCuda.CudaDNN
    Assembly: CudaDNN.dll
    Syntax
    public enum cudnnBatchNormMode

    Fields

    Name Description
    BatchNormPerActivation

    Normalization is performed per-activation. This mode is intended to be used after nonconvolutional network layers. In this mode bnBias and bnScale tensor dimensions are 1xCxHxW.

    BatchNormSpatial

    Normalization is performed over N+spatial dimensions. This mode is intended for use after convolutional layers (where spatial invariance is desired). In this mode bnBias, bnScale tensor dimensions are 1xCx1x1.

    BatchNormSpatialPersitent

    bnScale, bnBias tensor dims are 1xCx1x1 (one value per C-dim normalized over Nx1xHxW subtensors). May be faster than CUDNN_BATCHNORM_SPATIAL but imposes some limits on the range of values

    • Improve this Doc
    • View Source
    Back to top Generated by DocFX