Struct EinSumAnnotatedAsMatMul

pub struct EinSumAnnotatedAsMatMul<'a> {
    pub op: &'a EinSum,
    pub m_axis: &'a Axis,
    pub k_axis: &'a Axis,
    pub n_axis: &'a Axis,
    pub m: TDim,
    pub k: TDim,
    pub n: TDim,
}

Fields

op: &'a EinSum

m_axis: &'a Axis

k_axis: &'a Axis

n_axis: &'a Axis

m: TDim

k: TDim

n: TDim

Implementations

impl EinSumAnnotatedAsMatMul<'_>

pub fn a_m(&self) -> usize

pub fn a_k(&self) -> usize

pub fn b_k(&self) -> usize

pub fn b_n(&self) -> usize

pub fn c_m(&self) -> usize

pub fn c_n(&self) -> usize

Methods from Deref<Target = EinSum>

pub fn actual_input_shapes_from_facts<'m>( &self, inputs: &'m [impl Borrow<TypedFact>], ) -> Result<SmallVec<[Cow<'m, [TDim]>; 4]>, Error>

pub fn acceptable_accumulators(&self) -> SmallVec<[DatumType; 4]>

Trait Implementations

impl Debug for EinSumAnnotatedAsMatMul<'_>

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

Formats the value using the given formatter.

impl Deref for EinSumAnnotatedAsMatMul<'_>

type Target = EinSum

The resulting type after dereferencing.

fn deref(&self) -> &<EinSumAnnotatedAsMatMul<'_> as Deref>::Target

Dereferences the value.