[−][src]Trait tract_linalg::frame::mmm::MatMatMul

pub trait MatMatMul: Debug + Display + DynClone + Send + Sync + Any {
    pub fn a_pack(&self) -> Packer;

    pub fn b_pack(&self) -> Packer;

    pub fn a_storage(&self) -> &MatrixStoreSpec;

    pub fn b_storage(&self) -> &MatrixStoreSpec;

    pub fn c_storage(&self) -> &MatrixStoreSpec;

    pub fn internal_type(&self) -> DatumType;

    pub unsafe fn set_zero_point_a(&mut self, value: Tensor);

    pub unsafe fn set_zero_point_b(&mut self, value: Tensor);

    pub unsafe fn set_zero_point_c(&mut self, value: Tensor);

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

    pub unsafe fn b_from_data_and_offsets(
        &mut self, 
        rows_offsets: &[isize], 
        cols_offsets: &[isize]
    );

    pub unsafe fn b_vec_from_data_and_stride(&mut self, stride: isize);

    pub unsafe fn b_vec_from_data(&mut self);

    pub unsafe fn c_from_data_and_strides(
        &mut self, 
        row_stride: isize, 
        col_stride: isize
    );

    pub unsafe fn c_vec_from_data_and_stride(&mut self, stride: isize);

    pub unsafe fn c_vec_from_data(&mut self);

    pub unsafe fn run(
        &self, 
        a: &TensorView<'_>, 
        b: &TensorView<'_>, 
        c: &mut TensorView<'_>, 
        non_linear: &[FusedSpec]
    ) -> Result<()>;
}

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]

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]

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]

Loading content...

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]

Loading content...