torch_sys

Function at__amp_non_finite_check_and_unscale

source
pub unsafe extern "C" fn at__amp_non_finite_check_and_unscale(
    t: *mut C_tensor,
    found_inf: *mut C_tensor,
    inf_scale: *mut C_tensor,
)