Enum OpKind 

pub enum OpKind {
    Matmul {
        a: String,
        b: String,
    },
    Plus {
        a: String,
        b: String,
    },
    Hadamard {
        a: String,
        b: String,
    },
    Rmsnorm {
        x: String,
        residual: Option<String>,
        weight: Option<String>,
        heads: Option<usize>,
    },
    Rope {
        q: String,
        rc: String,
        rs: String,
        heads: Option<usize>,
    },
    Attention {
        q: String,
        k: String,
        v: String,
        heads: usize,
        groups: usize,
    },
    FullAttention {
        q: String,
        k: String,
        v: String,
        heads: usize,
    },
    Sigmoid {
        x: String,
    },
    Silu {
        x: String,
    },
    Gelu {
        x: String,
    },
    Softmax {
        x: String,
    },
    Transpose {
        x: String,
    },
    Layernorm {
        x: String,
        weight: String,
        bias: String,
    },
    CausalConv {
        x: String,
        weight: String,
        state: String,
        taps: usize,
    },
    L2norm {
        x: String,
        heads: Option<usize>,
    },
    Decay {
        ar: String,
        dt: String,
        alog: String,
    },
    GatedDelta {
        q: String,
        k: String,
        v: String,
        gate: String,
        beta: String,
        decay: String,
        state: String,
        norm: String,
        hv: usize,
        hk: usize,
    },
    Swiglu {
        gate: String,
        up: String,
    },
    Moe {
        x: String,
    },
    Argmax {
        x: String,
    },
    RowOf {
        table: String,
        tok: String,
    },
    SetRow {
        table: String,
        pos: String,
        row: String,
    },
}

Variants

Matmul

Fields

a: String

b: String

Plus

Fields

a: String

b: String

Hadamard

Fields

a: String

b: String

Rmsnorm

Fields

x: String

residual: Option<String>

weight: Option<String>

heads: Option<usize>

Rope

Fields

q: String

rc: String

rs: String

heads: Option<usize>

Attention

Fields

q: String

k: String

v: String

heads: usize

groups: usize

FullAttention

Fields

q: String

k: String

v: String

heads: usize

Sigmoid

Fields

x: String

Silu

Fields

x: String

Gelu

Fields

x: String

Softmax

Fields

x: String

Transpose

Fields

x: String

Layernorm

Fields

x: String

weight: String

bias: String

CausalConv

Fields

x: String

weight: String

state: String

taps: usize

L2norm

Fields

x: String

heads: Option<usize>

Decay

Fields

ar: String

dt: String

alog: String

GatedDelta

Fields

q: String

k: String

v: String

gate: String

beta: String

decay: String

state: String

norm: String

hv: usize

hk: usize

Swiglu

Fields

gate: String

up: String

Moe

Fields

x: String

Argmax

Fields

x: String

RowOf

Let V be row TOK of TABLE. — one row of a table by a data index (the embedding lookup).

Fields

table: String

tok: String

SetRow

Set row P of TABLE to ROW. — a scatter into a table (the KV-cache append).

Fields

table: String

pos: String

row: String

Implementations

impl OpKind

pub fn name(&self) -> &'static str

The op’s name — the key of torch_spelling.

pub fn inputs(&self) -> Vec<&str>

The variables the op reads.

pub fn writes(&self) -> Option<&str>

The carried state the op writes back (a conv ring, a GDN memory, a KV table), if any.

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

Formats the value using the given formatter.

impl PartialEq for OpKind

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for OpKind