Function torch_sys::c_generated::atg_cudnn_batch_norm_backward
source · [−]pub unsafe extern "C" fn atg_cudnn_batch_norm_backward(
out__: *mut *mut C_tensor,
input_: *mut C_tensor,
grad_output_: *mut C_tensor,
weight_: *mut C_tensor,
running_mean_: *mut C_tensor,
running_var_: *mut C_tensor,
save_mean_: *mut C_tensor,
save_var_: *mut C_tensor,
epsilon_: f64,
reserveSpace_: *mut C_tensor
)