[−]Trait tract_tensorflow::prelude::tract_linalg::frame::MatMatMul

pub trait MatMatMul: Send + Sync + Any + Debug + Display + DynClone {
    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<(), Error>;
}

Required methods

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<(), Error>

Loading content...

Implementors

impl<K, TA, TB, TC, TI> MatMatMul for MatMatMulImpl<K, TA, TB, TC, TI> where
    K: MatMatMulKer<TI> + 'static,
    TI: Datum + Copy + Add<TI> + Mul<TI, Output = TI> + Zero + Debug + 'static + Neg<Output = TI>,
    TA: Datum + Copy + Zero + Debug + 'static + AsPrimitive<TI>,
    TB: Datum + Copy + Zero + Debug + 'static + AsPrimitive<TI>,
    TC: Datum + Copy + Debug + 'static + Bounded + AsPrimitive<TI>,
    i32: AsPrimitive<TI>,
    usize: AsPrimitive<TI>, 

Loading content...