Struct EinSumAnnotatedAsLinear

pub struct EinSumAnnotatedAsLinear<'a> {
    pub op: &'a EinSum,
    pub m_axis: &'a Axis,
    pub k_axis: &'a Axis,
    pub n_axes: Vec<&'a Axis>,
    pub m: usize,
    pub k: usize,
    pub ns: Vec<&'a TDim>,
    pub act_dt: DatumType,
    pub weight_type: WeightType,
}

Fields

op: &'a EinSum

m_axis: &'a Axis

k_axis: &'a Axis

n_axes: Vec<&'a Axis>

m: usize

k: usize

ns: Vec<&'a TDim>

act_dt: DatumType

weight_type: WeightType

Implementations

impl<'a> EinSumAnnotatedAsLinear<'a>

pub fn from( model: &'a Graph<TypedFact, Box<dyn TypedOp>>, node: &'a Node<TypedFact, Box<dyn TypedOp>>, op: &'a EinSum, ) -> Result<Option<EinSumAnnotatedAsLinear<'a>>, Error>

pub fn weight_m_axis(&self) -> usize

pub fn weight_k_axis(&self) -> usize

pub fn input_k_axis(&self) -> usize

pub fn output_m_axis(&self) -> usize

pub fn need_mmv(&self) -> bool

pub fn need_mmm(&self) -> bool

pub fn cost_for_weights( &self, format: &(dyn MMMInputFormat + 'static), ) -> Option<usize>

pub fn preferred_packing(&self) -> Box<dyn MMMInputFormat>

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 EinSumAnnotatedAsLinear<'_>

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

Formats the value using the given formatter.

impl Deref for EinSumAnnotatedAsLinear<'_>

type Target = EinSum

The resulting type after dereferencing.

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

Dereferences the value.