[−][src]Trait tract_linalg::frame::MatMatMul
Required methods
pub fn a_pack(&self) -> Packer
[src]
pub fn b_pack(&self) -> Packer
[src]
pub fn a_storage(&self) -> &MatrixStoreSpec
[src]
pub fn b_storage(&self) -> &MatrixStoreSpec
[src]
pub fn c_storage(&self) -> &MatrixStoreSpec
[src]
pub fn internal_type(&self) -> DatumType
[src]
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]
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<()>
Implementors
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<()>