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,
)