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

pub trait MatMatMul: Debug + Display + DynClone + Send + Sync + Any {

Show 16 methods    fn kernel_name(&self) -> &'static str;

    fn mr(&self) -> usize;

    fn nr(&self) -> usize;

    fn a_pack(&self, k: usize) -> Packer;

    fn b_pack(&self, k: usize) -> Packer;

    fn internal_type(&self) -> DatumType;

    unsafe fn a_packed(&self, item_size: usize, k: usize) -> MatrixStoreSpec;

    unsafe fn b_packed(&self, item_size: usize, k: usize) -> MatrixStoreSpec;

    unsafe fn b_from_data_and_offsets(
        &self, 
        item_size: usize, 
        rows_offsets: &[isize], 
        cols_offsets: &[isize]
    ) -> MatrixStoreSpec;

    unsafe fn c_view(&self) -> MatrixStoreSpec;

    unsafe fn c_view_with_axis(
        &self, 
        m_axis: usize, 
        n_axis: usize
    ) -> MatrixStoreSpec;

    unsafe fn c_from_data_and_strides(
        &self, 
        item_size: usize, 
        row_stride: isize, 
        col_stride: isize
    ) -> MatrixStoreSpec;

    unsafe fn allocate_scratch_space(&self) -> Box<dyn ScratchSpace>;

    unsafe fn can_use_scratch_space(&self, scratch: &dyn ScratchSpace) -> bool;

    unsafe fn run_with_scratch_space(
        &self, 
        m: usize, 
        k: usize, 
        n: usize, 
        scratch: &mut dyn ScratchSpace, 
        a: &MatrixStore<'_, '_>, 
        b: &MatrixStore<'_, '_>, 
        c: &mut MatrixStore<'_, '_>, 
        non_linear: &[FusedSpec<'_>]
    ) -> Result<()>;

    unsafe fn run(
        &self, 
        m: usize, 
        k: usize, 
        n: usize, 
        a: &MatrixStore<'_, '_>, 
        b: &MatrixStore<'_, '_>, 
        c: &mut MatrixStore<'_, '_>, 
        non_linear: &[FusedSpec<'_>]
    ) -> Result<()> { ... }

}

Required methods

Provided methods

Implementors