pub unsafe extern "C" fn atg__batch_norm_with_update(
out__: *mut *mut C_tensor,
input_: *mut C_tensor,
weight_: *mut C_tensor,
bias_: *mut C_tensor,
running_mean_: *mut C_tensor,
running_var_: *mut C_tensor,
momentum_: f64,
eps_: f64,
)