pub unsafe extern "C" fn atg_batch_norm_gather_stats(
out__: *mut *mut C_tensor,
input_: *mut C_tensor,
mean_: *mut C_tensor,
invstd_: *mut C_tensor,
running_mean_: *mut C_tensor,
running_var_: *mut C_tensor,
momentum_: f64,
eps_: f64,
count_: i64,
)