torch_sys::c_generated

Function atg__scaled_mm

source
pub unsafe extern "C" fn atg__scaled_mm(
    out__: *mut *mut C_tensor,
    self_: *mut C_tensor,
    mat2_: *mut C_tensor,
    scale_a_: *mut C_tensor,
    scale_b_: *mut C_tensor,
    bias_: *mut C_tensor,
    scale_result_: *mut C_tensor,
    out_dtype_: c_int,
    use_fast_accum_: c_int,
)