Function atg_batch_norm

pub unsafe extern "C" fn atg_batch_norm(
    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,
    training_: c_int,
    momentum_: f64,
    eps_: f64,
    cudnn_enabled_: c_int,
)