Enum OpKind 

pub enum OpKind {
    Add,
    Mul,
    Sub,
    Div,
    Relu,
    Sigmoid,
    Tanh,
    Scale,
    Axpy,
    MatMul,
    Reduce,
    Transpose,
}

The kind of an operation in the fusion graph.

The first nine variants are elementwise and therefore fusible; the final three are barrier ops that change the iteration space or reduce/permute data and so cannot participate in an elementwise fusion group.

Variants

Add

Elementwise addition.

Mul

Elementwise multiplication.

Sub

Elementwise subtraction.

Div

Elementwise division.

Relu

Rectified linear unit activation.

Sigmoid

Logistic sigmoid activation.

Tanh

Hyperbolic tangent activation.

Scale

Scalar scaling (alpha * x).

Axpy

Fused multiply-add (alpha * x + y).

MatMul

Matrix multiplication (barrier: not fusible into an elementwise group).

Reduce

Reduction such as sum/mean/max (barrier).

Transpose

Transpose / permutation (barrier).

Implementations

impl OpKind

pub fn is_fusible(&self) -> bool

Returns true if the op is an elementwise op that may be fused into a single kernel together with other fusible ops.

pub fn is_barrier(&self) -> bool

Returns true for barrier ops (MatMul, Reduce, Transpose) that break fusion chains and always form singleton groups.

Trait Implementations

impl Clone for OpKind

fn clone(&self) -> OpKind

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for OpKind

impl Debug for OpKind

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

Formats the value using the given formatter.

impl Eq for OpKind

impl Hash for OpKind

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for OpKind

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for OpKind