[−][src]Struct tract_linalg::frame::mmm::MatMatMulImpl
Fields
m: usize
k: usize
n: usize
a_storage: MatrixStoreSpec
b_storage: MatrixStoreSpec
c_storage: MatrixStoreSpec
zero_point_a: Option<Tensor>
zero_point_b: Option<Tensor>
zero_point_c: Option<Tensor>
scale_factor: Option<(TI, usize)>
Implementations
impl<K, TA, TB, TC, TI> MatMatMulImpl<K, TA, TB, TC, TI> where
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
[src]
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
Trait Implementations
impl<K: Clone, TA: Clone, TB: Clone, TC: Clone, TI: Clone> Clone for MatMatMulImpl<K, TA, TB, TC, TI> where
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
[src]
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
pub fn clone(&self) -> MatMatMulImpl<K, TA, TB, TC, TI>
[src]
pub fn clone_from(&mut self, source: &Self)
1.0.0[src]
impl<K: Debug, TA: Debug, TB: Debug, TC: Debug, TI: Debug> Debug for MatMatMulImpl<K, TA, TB, TC, TI> where
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
[src]
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
impl<K, TA, TB, TC, TI> Display for MatMatMulImpl<K, TA, TB, TC, TI> where
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI>,
[src]
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI>,
impl<K, TA, TB, TC, TI> MatMatMul for MatMatMulImpl<K, TA, TB, TC, TI> where
TA: Datum + Copy + Zero + Debug + 'static + AsPrimitive<TI>,
TB: Datum + Copy + Zero + Debug + 'static + AsPrimitive<TI>,
TC: Datum + Copy + Debug + 'static + Bounded + AsPrimitive<TI>,
TI: Datum + Copy + Add + Mul<Output = TI> + Zero + Debug + 'static + Neg<Output = TI>,
K: MatMatMulKer<TI> + 'static,
i32: AsPrimitive<TI>,
usize: AsPrimitive<TI>,
[src]
TA: Datum + Copy + Zero + Debug + 'static + AsPrimitive<TI>,
TB: Datum + Copy + Zero + Debug + 'static + AsPrimitive<TI>,
TC: Datum + Copy + Debug + 'static + Bounded + AsPrimitive<TI>,
TI: Datum + Copy + Add + Mul<Output = TI> + Zero + Debug + 'static + Neg<Output = TI>,
K: MatMatMulKer<TI> + 'static,
i32: AsPrimitive<TI>,
usize: AsPrimitive<TI>,
pub fn a_pack(&self) -> Packer
[src]
pub fn b_pack(&self) -> Packer
[src]
pub fn internal_type(&self) -> DatumType
[src]
pub fn a_storage(&self) -> &MatrixStoreSpec
[src]
pub fn b_storage(&self) -> &MatrixStoreSpec
[src]
pub fn c_storage(&self) -> &MatrixStoreSpec
[src]
pub unsafe fn b_from_data_and_offsets(
&mut self,
rows_offsets: &[isize],
cols_offsets: &[isize]
)
[src]
&mut self,
rows_offsets: &[isize],
cols_offsets: &[isize]
)
pub unsafe fn b_vec_from_data_and_stride(&mut self, stride: isize)
[src]
pub unsafe fn b_vec_from_data(&mut self)
[src]
pub unsafe fn c_from_data_and_strides(
&mut self,
row_stride: isize,
col_stride: isize
)
[src]
&mut self,
row_stride: isize,
col_stride: isize
)
pub unsafe fn c_vec_from_data_and_stride(&mut self, stride: isize)
[src]
pub unsafe fn c_vec_from_data(&mut self)
[src]
pub unsafe fn run(
&self,
a: &TensorView<'_>,
b: &TensorView<'_>,
c: &mut TensorView<'_>,
non_linear: &[FusedSpec]
) -> Result<()>
[src]
&self,
a: &TensorView<'_>,
b: &TensorView<'_>,
c: &mut TensorView<'_>,
non_linear: &[FusedSpec]
) -> Result<()>
pub unsafe fn set_zero_point_a(&mut self, value: Tensor)
[src]
pub unsafe fn set_zero_point_b(&mut self, value: Tensor)
[src]
pub unsafe fn set_zero_point_c(&mut self, value: Tensor)
[src]
pub unsafe fn set_scale_factor(&mut self, factor: f32)
[src]
impl<K, TA, TB, TC, TI> Send for MatMatMulImpl<K, TA, TB, TC, TI> where
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
[src]
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
impl<K, TA, TB, TC, TI> Sync for MatMatMulImpl<K, TA, TB, TC, TI> where
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
[src]
TA: Copy + Zero + 'static,
TB: Copy + Zero + 'static,
TC: Copy + Debug + 'static,
TI: Copy + Add + Mul + Zero + Debug + 'static,
K: MatMatMulKer<TI> + 'static,
Auto Trait Implementations
impl<K, TA, TB, TC, TI> RefUnwindSafe for MatMatMulImpl<K, TA, TB, TC, TI> where
K: RefUnwindSafe,
TA: RefUnwindSafe,
TB: RefUnwindSafe,
TC: RefUnwindSafe,
TI: RefUnwindSafe,
K: RefUnwindSafe,
TA: RefUnwindSafe,
TB: RefUnwindSafe,
TC: RefUnwindSafe,
TI: RefUnwindSafe,
impl<K, TA, TB, TC, TI> Unpin for MatMatMulImpl<K, TA, TB, TC, TI> where
K: Unpin,
TA: Unpin,
TB: Unpin,
TC: Unpin,
TI: Unpin,
K: Unpin,
TA: Unpin,
TB: Unpin,
TC: Unpin,
TI: Unpin,
impl<K, TA, TB, TC, TI> UnwindSafe for MatMatMulImpl<K, TA, TB, TC, TI> where
K: UnwindSafe,
TA: UnwindSafe,
TB: UnwindSafe,
TC: UnwindSafe,
TI: UnwindSafe,
K: UnwindSafe,
TA: UnwindSafe,
TB: UnwindSafe,
TC: UnwindSafe,
TI: UnwindSafe,
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized,
[src]
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
pub fn borrow_mut(&mut self) -> &mut T
[src]
impl<T> DynClone for T where
T: Clone,
[src]
T: Clone,
pub fn __clone_box(&self, Private) -> *mut ()
[src]
impl<T> From<T> for T
[src]
impl<T, U> Into<U> for T where
U: From<T>,
[src]
U: From<T>,
impl<T> ToOwned for T where
T: Clone,
[src]
T: Clone,
type Owned = T
The resulting type after obtaining ownership.
pub fn to_owned(&self) -> T
[src]
pub fn clone_into(&self, target: &mut T)
[src]
impl<T> ToString for T where
T: Display + ?Sized,
[src]
T: Display + ?Sized,
impl<T, U> TryFrom<U> for T where
U: Into<T>,
[src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
pub fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>
[src]
impl<T, U> TryInto<U> for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,