Struct MatmulTransformAnalysis 

pub struct MatmulTransformAnalysis { /* private fields */ }

The facts of a batched matmul lhs @ rhs relevant to folding its batches into its rows.

Implementations

impl MatmulTransformAnalysis

pub fn from_shapes(lhs: &Shape, rhs: &Shape) -> Self

Analyze from shapes alone: the operands are assumed batch-contiguous, as holds for freshly materialized tensors. Use Self::from_metadata when the layouts are known.

The rhs may have a lower rank than the lhs (implicit broadcast).

pub fn from_metadata(lhs: &Metadata, rhs: &Metadata, out: &Metadata) -> Self

Analyze from full metadata: on top of the shape facts, the lhs and out rows must advance with a single stride across batch boundaries.

Trait Implementations

impl Clone for MatmulTransformAnalysis

fn clone(&self) -> MatmulTransformAnalysis

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for MatmulTransformAnalysis

impl Debug for MatmulTransformAnalysis

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

Formats the value using the given formatter.

impl Eq for MatmulTransformAnalysis

impl PartialEq for MatmulTransformAnalysis

fn eq(&self, other: &MatmulTransformAnalysis) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for MatmulTransformAnalysis