Enum tract_linalg::frame::mmm::FusedSpec

pub enum FusedSpec<'t> {
    BinScalar(&'t Tensor, BinOp),
    BinPerRow(TensorView<'t>, BinOp),
    BinPerCol(TensorView<'t>, BinOp),
    AddRowColProducts(&'t Tensor, &'t Tensor),
    AddUnicast(OutputStore),
    LeakyRelu(&'t Tensor),
    QScale(isize, RoundingPolicy, i32),
    RoundingShiftRight(usize, RoundingPolicy),
    ShiftLeft(usize),
    Store(OutputStore),
    AddMatMul {
        a: &'t dyn MMMInput,
        b: &'t dyn MMMInput,
    },
}

Variants

BinScalar(&'t Tensor, BinOp)

BinPerRow(TensorView<'t>, BinOp)

BinPerCol(TensorView<'t>, BinOp)

AddRowColProducts(&'t Tensor, &'t Tensor)

AddUnicast(OutputStore)

LeakyRelu(&'t Tensor)

QScale(isize, RoundingPolicy, i32)

RoundingShiftRight(usize, RoundingPolicy)

ShiftLeft(usize)

Store(OutputStore)

AddMatMul

Fields

a: &'t dyn MMMInput

b: &'t dyn MMMInput

Implementations

impl<'t> FusedSpec<'t>

pub fn prefer_col_outer(&self) -> bool

Trait Implementations

impl<'t> Clone for FusedSpec<'t>

fn clone(&self) -> FusedSpec<'t>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<'t> Debug for FusedSpec<'t>

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

Formats the value using the given formatter.