torch_sys::c_generated

Function atg_batch_norm_backward_reduce

source
pub unsafe extern "C" fn atg_batch_norm_backward_reduce(
    out__: *mut *mut C_tensor,
    grad_out_: *mut C_tensor,
    input_: *mut C_tensor,
    mean_: *mut C_tensor,
    invstd_: *mut C_tensor,
    weight_: *mut C_tensor,
    input_g_: c_int,
    weight_g_: c_int,
    bias_g_: c_int,
)