Enum ProtoFusedSpec

pub enum ProtoFusedSpec {
    AddMatMul {
        geo: AddMatMulGeometry,
        a: usize,
        b: usize,
        packings: Vec<(usize, Option<PanelExtractor>)>,
    },
    BinScalar(usize, BinOp),
    LeakyRelu(usize),
    BinPerRow(usize, BinOp, MapOutputAxisToInput),
    BinPerCol(usize, BinOp, MapOutputAxisToInput),
    AddRowColProducts(usize, usize),
    AddUnicast(OutputStoreSpec, usize, MapOutputAxisToInput),
    Scaler(Scaler),
    Store(Vec<OutputStoreSpec>),
}

Variants

AddMatMul

Fields

geo: AddMatMulGeometry

a: usize

b: usize

packings: Vec<(usize, Option<PanelExtractor>)>

BinScalar(usize, BinOp)

LeakyRelu(usize)

BinPerRow(usize, BinOp, MapOutputAxisToInput)

BinPerCol(usize, BinOp, MapOutputAxisToInput)

AddRowColProducts(usize, usize)

AddUnicast(OutputStoreSpec, usize, MapOutputAxisToInput)

Scaler(Scaler)

Store(Vec<OutputStoreSpec>)

Implementations

impl ProtoFusedSpec

pub fn format(&self, mmm: &(dyn MatMatMul + 'static), mode: usize) -> String

pub fn resolve<'t>( &'t self, inputs: &'t [TValue], output_coords: &[usize], output: &Tensor, mmm: &(dyn MatMatMul + 'static), mode: usize, ) -> FusedSpec<'t>

pub fn is_trivial(&self) -> bool

pub fn resolve_trivial<'t>( &'t self, inputs: &'t [TValue], output: &mut Tensor, _mmm: &(dyn MatMatMul + 'static), mode: usize, ) -> FusedSpec<'t>

Trait Implementations

impl Clone for ProtoFusedSpec

fn clone(&self) -> ProtoFusedSpec

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ProtoFusedSpec

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

Formats the value using the given formatter.