pub unsafe extern "C" fn atg_batch_norm_gather_stats_with_counts_out(
out__: *mut *mut C_tensor,
out0_: *mut C_tensor,
out1_: *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,
counts_: *mut C_tensor
)