[−][src]Trait tract_linalg::frame::mmm::QMatMatMul
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>]
)
&self,
a: *const TA,
b: *const TB,
c: *mut TC,
non_linear: &[FusedSpec<TI>]
)
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]
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>,
fn as_mmm(&self) -> &dyn MatMatMul<TA, TB, TC, TI>
[src]
fn as_mmm_mut(&mut self) -> &mut dyn MatMatMul<TA, TB, TC, TI>
[src]
unsafe fn set_zero_point_a_scalar(&mut self, value: TA)
[src]
unsafe fn set_zero_point_b_scalar(&mut self, value: TB)
[src]
unsafe fn set_zero_point_c_scalar(&mut self, value: TC)
[src]
unsafe fn set_zero_point_a_vector(&mut self, values: Vec<TA>)
[src]
unsafe fn set_zero_point_b_vector(&mut self, values: Vec<TB>)
[src]
unsafe fn set_scale_factor(&mut self, factor: f32)
[src]
unsafe fn run(
&self,
a: *const TA,
b: *const TB,
c: *mut TC,
non_linear: &[FusedSpec<TI>]
)
[src]
&self,
a: *const TA,
b: *const TB,
c: *mut TC,
non_linear: &[FusedSpec<TI>]
)