Enum OpKind 

pub enum OpKind {
    Embedding,
    Range,
    Neg,
    Relu,
    Gelu,
    Silu,
    Sigmoid,
    Tanh,
    Exp,
    Log,
    Sqrt,
    Transpose,
    Add,
    Sub,
    Mul,
    Div,
    Mod,
    Pow,
    MatMul,
    Sum {
        dims: Vec<i64>,
        keepdim: bool,
    },
    Mean {
        dims: Vec<i64>,
        keepdim: bool,
    },
    Max {
        dim: i64,
        keepdim: bool,
    },
    Min {
        dim: i64,
        keepdim: bool,
    },
    Variance {
        dims: Vec<i64>,
        keepdim: bool,
    },
    LayerNorm {
        eps: f64,
    },
    BatchNorm {
        eps: f64,
    },
    MultiHeadAttention {
        n_heads: i64,
    },
    TransformerBlock {
        n_heads: i64,
    },
    Softmax {
        dim: i64,
    },
    Reshape {
        target_shape: Vec<Dim>,
    },
    View {
        target_shape: Vec<Dim>,
    },
    Permute {
        dims: Vec<i64>,
    },
    Concat {
        dim: i64,
    },
    Split {
        dim: i64,
        chunks: i64,
    },
    Expand {
        target_shape: Vec<Dim>,
    },
    Dropout {
        p: f64,
    },
    Linear {
        bias: bool,
    },
    CrossEntropy,
    MseLoss,
    Equal,
    NotEqual,
    Less,
    Greater,
    LessEqual,
    GreaterEqual,
    And,
    Or,
    Not,
    Constant(ConstantValue),
    Repeat {
        count: i64,
        body_op: Box<OpKind>,
    },
    Custom {
        name: String,
        attrs: HashMap<String, AttrValue>,
    },
    Call {
        graph_name: String,
    },
    Identity,
}

An operation in the computation graph.

Variants

Embedding

Look up rows in an embedding table.

Range

Generate a range of values.

Neg

Relu

Gelu

Silu

Sigmoid

Tanh

Exp

Log

Sqrt

Transpose

Add

Sub

Mul

Div

Mod

Pow

MatMul

Sum

Fields

dims: Vec<i64>

keepdim: bool

Mean

Fields

dims: Vec<i64>

keepdim: bool

Max

Fields

dim: i64

keepdim: bool

Min

Fields

dim: i64

keepdim: bool

Variance

Fields

dims: Vec<i64>

keepdim: bool

LayerNorm

Fields

eps: f64

BatchNorm

Fields

eps: f64

MultiHeadAttention

Fields

n_heads: i64

TransformerBlock

Fields

n_heads: i64

Softmax

Fields

dim: i64

Reshape

Fields

target_shape: Vec<Dim>

View

Fields

target_shape: Vec<Dim>

Permute

Fields

dims: Vec<i64>

Concat

Fields

dim: i64

Split

Fields

dim: i64

chunks: i64

Expand

Fields

target_shape: Vec<Dim>

Dropout

Fields

p: f64

Linear

Fields

bias: bool

CrossEntropy

MseLoss

Equal

NotEqual

Less

Greater

LessEqual

GreaterEqual

And

Or

Not

Constant(ConstantValue)

Repeat

Fields

count: i64

body_op: Box<OpKind>

Custom

Fields

name: String

attrs: HashMap<String, AttrValue>

Call

Fields

graph_name: String

Identity

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 Debug for OpKind

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

Formats the value using the given formatter.

impl Display for OpKind

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

Formats the value using the given formatter.