Function rcudnn::cudnnGetBatchNormalizationForwardTrainingExWorkspaceSize[][src]

pub unsafe extern "C" fn cudnnGetBatchNormalizationForwardTrainingExWorkspaceSize(
    handle: *mut cudnnContext,
    mode: cudnnBatchNormMode_t,
    bnOps: cudnnBatchNormOps_t,
    xDesc: *mut cudnnTensorStruct,
    zDesc: *mut cudnnTensorStruct,
    yDesc: *mut cudnnTensorStruct,
    bnScaleBiasMeanVarDesc: *mut cudnnTensorStruct,
    activationDesc: *mut cudnnActivationStruct,
    sizeInBytes: *mut usize
) -> cudnnStatus_t