Function torch_sys::c_generated::atg_batch_norm_gather_stats_out
source · pub unsafe extern "C" fn atg_batch_norm_gather_stats_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,
count_: i64
)