Struct MatMulInput 

pub struct MatMulInput {
    pub a: Vec<f32>,
    pub b: Vec<f32>,
    pub rows_a: usize,
    pub cols_a: usize,
    pub cols_b: usize,
}

Input for matrix multiplication.

Fields

a: Vec<f32>

First matrix (row-major, dimensions rows_a x cols_a).

b: Vec<f32>

Second matrix (row-major, dimensions cols_a x cols_b).

rows_a: usize

Rows in matrix A.

cols_a: usize

Columns in matrix A (= rows in matrix B).

cols_b: usize

Columns in matrix B.

Trait Implementations

impl BatchKernel<MatMulInput, MatMulOutput> for MatMul

fn execute<'life0, 'async_trait>( &'life0 self, input: MatMulInput, ) -> Pin<Box<dyn Future<Output = Result<MatMulOutput>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Execute the kernel with the given input.

fn validate_input(&self, input: &MatMulInput) -> Result<()>

Validate the input before execution.

fn execute_with_context<'life0, 'life1, 'async_trait>( &'life0 self, ctx: &'life1 ExecutionContext, input: I, ) -> Pin<Box<dyn Future<Output = Result<O>> + Send + 'async_trait>>

where I: 'async_trait, Self: Sync + 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Execute the kernel with execution context.

fn execute_with_timeout<'life0, 'async_trait>( &'life0 self, input: I, timeout: Duration, ) -> Pin<Box<dyn Future<Output = Result<O>> + Send + 'async_trait>>

where I: 'async_trait, Self: Sync + 'async_trait, 'life0: 'async_trait,

Execute the kernel with a timeout.

impl Clone for MatMulInput

fn clone(&self) -> MatMulInput

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MatMulInput

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for MatMulInput

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for MatMulInput

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.