torch_sys::c_generated

Function atg_cudnn_batch_norm_out

source
pub unsafe extern "C" fn atg_cudnn_batch_norm_out(
    out__: *mut *mut C_tensor,
    out0_: *mut C_tensor,
    out1_: *mut C_tensor,
    out2_: *mut C_tensor,
    out3_: *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,
    exponential_average_factor_: f64,
    epsilon_: f64,
)