Enum kn_graph::graph::Operation

pub enum Operation {
    Input {
        index: usize,
    },
    Constant {
        tensor: DTensor,
    },
    View {
        input: Value,
    },
    Broadcast {
        input: Value,
    },
    Permute {
        input: Value,
        permutation: Vec<usize>,
    },
    Slice {
        input: Value,
        axis: usize,
        range: SliceRange,
    },
    Flip {
        input: Value,
        axis: usize,
    },
    Gather {
        input: Value,
        axis: usize,
        indices: Value,
    },
    Concat {
        inputs: Vec<Value>,
        axis: usize,
    },
    Conv {
        input: Value,
        filter: Value,
        details: ConvDetails,
    },
    MatMul {
        left: Value,
        right: Value,
    },
    Unary {
        input: Value,
        op: UnaryOp,
    },
    Binary {
        left: Value,
        right: Value,
        op: BinaryOp,
    },
    Softmax {
        input: Value,
        axis: usize,
    },
    Layernorm {
        input: Value,
        axis: usize,
        eps: Total<f32>,
    },
    Reduce {
        input: Value,
        axes: Vec<usize>,
        op: ReduceOp,
    },
}

The core set of graph operations. Some attempt was made to keep operations orthogonal but flexible, so they can be composed easily.

Variants

Input

Fields

index: usize

A runtime-variable input.

Constant

Fields

tensor: DTensor

A constant built into the network.

View

Fields

input: Value

View a value as a different shape.

Broadcast

Fields

input: Value

Repeat along all axes with size 1 that don’t match the output shape.

Permute

Fields

input: Value

permutation: Vec<usize>

Change the order of axis in the shape.

Slice

Fields

input: Value

axis: usize

range: SliceRange

Slice along the given axis with range start..end.

Flip

Fields

input: Value

axis: usize

Flip the given axis.

Gather

Fields

input: Value

axis: usize

indices: Value

Gather values from input at the indices in index on the given axis. indices is a rank-1 tensor.

Concat

Fields

inputs: Vec<Value>

axis: usize

Concatenate values along an axis.

Conv

Fields

input: Value

filter: Value

details: ConvDetails

2D convolution.

MatMul

Fields

left: Value

right: Value

(Batched) Matrix multiply. If left has shape [b, p, q] and right has shape [b, q, r] the result has shape [b, p, r].

Unary

Fields

input: Value

op: UnaryOp

Elementwise unary operation.

Binary

Fields

left: Value

right: Value

op: BinaryOp

Elementwise binary operation. Both operands must have the same shape.

Softmax

Fields

input: Value

axis: usize

Softmax along axis.

Layernorm

Fields

input: Value

axis: usize

eps: Total<f32>

Layernorm along axis.

Reduce

Fields

input: Value

axes: Vec<usize>

op: ReduceOp

Reduce along the given axes using op. The axes are removed from the shape.

Implementations

impl Operation

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

Trait Implementations

impl Clone for Operation

fn clone(&self) -> Operation

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Operation

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

Formats the value using the given formatter.

impl PartialEq for Operation

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

This method tests for self and other values to be equal, and is used by ==.

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

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

impl Eq for Operation

impl StructuralEq for Operation

impl StructuralPartialEq for Operation