pub unsafe extern "C" fn atg_batch_norm_update_stats( out__: *mut *mut C_tensor, input_: *mut C_tensor, running_mean_: *mut C_tensor, running_var_: *mut C_tensor, momentum_: f64, )