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)