Function rcudnn::cudnnGetNormalizationBackwardWorkspaceSize[][src]

pub unsafe extern "C" fn cudnnGetNormalizationBackwardWorkspaceSize(
    handle: *mut cudnnContext,
    mode: cudnnNormMode_t,
    normOps: cudnnNormOps_t,
    algo: cudnnNormAlgo_t,
    xDesc: *mut cudnnTensorStruct,
    yDesc: *mut cudnnTensorStruct,
    dyDesc: *mut cudnnTensorStruct,
    dzDesc: *mut cudnnTensorStruct,
    dxDesc: *mut cudnnTensorStruct,
    dNormScaleBiasDesc: *mut cudnnTensorStruct,
    activationDesc: *mut cudnnActivationStruct,
    normMeanVarDesc: *mut cudnnTensorStruct,
    sizeInBytes: *mut usize,
    groupCnt: i32
) -> cudnnStatus_t