Struct MatmulProblem

pub struct MatmulProblem {
    pub m: usize,
    pub n: usize,
    pub k: usize,
    pub lhs_batches: Vec<usize>,
    pub rhs_batches: Vec<usize>,
    pub lhs_layout: MatrixLayout,
    pub rhs_layout: MatrixLayout,
}

Description of a matmul problem to solve, regardless of actual data

Fields

m: usize

Number of rows in the output matrix

n: usize

Number of columns in the output matrix

k: usize

Reduction dimension

lhs_batches: Vec<usize>

Batch shape for Lhs tensor

rhs_batches: Vec<usize>

Batch shape for Rhs tensor

lhs_layout: MatrixLayout

Memory layout of the Lhs matrix.

rhs_layout: MatrixLayout

Memory layout of the Rhs matrix.

Trait Implementations

impl Clone for MatmulProblem

fn clone(&self) -> MatmulProblem

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MatmulProblem

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

Formats the value using the given formatter.

impl From<&MatmulProblem> for MatmulKind

fn from(problem: &MatmulProblem) -> Self

Converts to this type from the input type.

impl From<MatmulProblem> for MatmulKind

fn from(problem: MatmulProblem) -> Self

Converts to this type from the input type.

impl From<MatmulProblem> for MatmulProblemSize

fn from(problem: MatmulProblem) -> Self

Converts to this type from the input type.