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