[−][src]Trait tract_linalg::frame::QMatMatMul

pub trait QMatMatMul<TA, TB, TC, TI>: Debug + Display + DynClone + Send + Sync + DynHash where
    TA: Copy + Zero + 'static,
    TB: Copy + Zero + 'static,
    TC: Copy + Debug + 'static,
    TI: Copy + Add + Mul + Zero + Debug + 'static, {
    fn as_mmm(&self) -> &dyn MatMatMul<TA, TB, TC, TI>;

    fn as_mmm_mut(&mut self) -> &mut dyn MatMatMul<TA, TB, TC, TI>;

    unsafe fn set_zero_point_a_scalar(&mut self, value: TA);

    unsafe fn set_zero_point_a_vector(&mut self, values: Vec<TA>);

    unsafe fn set_zero_point_b_scalar(&mut self, value: TB);

    unsafe fn set_zero_point_b_vector(&mut self, values: Vec<TB>);

    unsafe fn set_zero_point_c_scalar(&mut self, value: TC);

    unsafe fn set_scale_factor(&mut self, factor: f32);

    unsafe fn run(
        &self, 
        a: *const TA, 
        b: *const TB, 
        c: *mut TC, 
        non_linear: &[FusedSpec<TI>]
    );
}

Required methods

fn as_mmm(&self) -> &dyn MatMatMul<TA, TB, TC, TI>

fn as_mmm_mut(&mut self) -> &mut dyn MatMatMul<TA, TB, TC, TI>

unsafe fn set_zero_point_a_scalar(&mut self, value: TA)

unsafe fn set_zero_point_a_vector(&mut self, values: Vec<TA>)

unsafe fn set_zero_point_b_scalar(&mut self, value: TB)

unsafe fn set_zero_point_b_vector(&mut self, values: Vec<TB>)

unsafe fn set_zero_point_c_scalar(&mut self, value: TC)

unsafe fn set_scale_factor(&mut self, factor: f32)

unsafe fn run(
    &self,
    a: *const TA,
    b: *const TB,
    c: *mut TC,
    non_linear: &[FusedSpec<TI>]
)

Loading content...

Implementors

impl<K, TA, TB, TC, TI> QMatMatMul<TA, TB, TC, TI> for QMatMatMulImpl<K, TA, TB, TC, TI> where
    TA: Copy + Zero + Debug + SloppyHash + AsPrimitive<TI> + 'static,
    TB: Copy + Zero + Debug + SloppyHash + AsPrimitive<TI> + 'static,
    TC: Copy + Debug + Bounded + AsPrimitive<TI> + SloppyHash + 'static,
    TI: Copy + Add + Mul<Output = TI> + Zero + Neg<Output = TI> + Debug + SloppyHash + 'static,
    K: MatMatMulKer<TA, TB, TC, TI> + 'static,
    usize: AsPrimitive<TI>,
    i32: AsPrimitive<TI>, [src]

Loading content...