torch_sys::c_generated

Function atg__amp_update_scale_out

source
pub unsafe extern "C" fn atg__amp_update_scale_out(
    out__: *mut *mut C_tensor,
    out_: *mut C_tensor,
    self_: *mut C_tensor,
    growth_tracker_: *mut C_tensor,
    found_inf_: *mut C_tensor,
    scale_growth_factor_: f64,
    scale_backoff_factor_: f64,
    growth_interval_: i64,
)