Function rcudnn::cudnnBatchNormalizationForwardInference
source · pub unsafe extern "C" fn cudnnBatchNormalizationForwardInference(
handle: *mut cudnnContext,
mode: cudnnBatchNormMode_t,
alpha: *const c_void,
beta: *const c_void,
xDesc: *mut cudnnTensorStruct,
x: *const c_void,
yDesc: *mut cudnnTensorStruct,
y: *mut c_void,
bnScaleBiasMeanVarDesc: *mut cudnnTensorStruct,
bnScale: *const c_void,
bnBias: *const c_void,
estimatedMean: *const c_void,
estimatedVariance: *const c_void,
epsilon: f64
) -> cudnnStatus_t