Enum Op

pub enum Op<T: DType> {
    Fill {
        v: T,
    },
    Arange {
        start: T,
        step: T,
        stop: T,
    },
    BinaryOp {
        l_id: GraphTensorId,
        r_id: GraphTensorId,
        operator: BinaryOpType,
    },
    UnaryOp {
        v_id: GraphTensorId,
        operator: UnaryOpType,
    },
    FusedMulAdd {
        a_id: GraphTensorId,
        b_id: GraphTensorId,
        c_id: GraphTensorId,
    },
    MatMul {
        l_id: GraphTensorId,
        r_id: GraphTensorId,
        o_id: Option<GraphTensorId>,
        k: usize,
        alpha: T,
        beta: T,
    },
    Rand,
    Randn {
        mean: T,
        std: T,
    },
    Permute {
        v_id: GraphTensorId,
    },
    NoOp,
}

Variants

Fill

Fields

v: T

Arange

Fields

start: T

step: T

stop: T

BinaryOp

Fields

l_id: GraphTensorId

r_id: GraphTensorId

operator: BinaryOpType

UnaryOp

Fields

v_id: GraphTensorId

operator: UnaryOpType

FusedMulAdd

a * b + c

Fields

a_id: GraphTensorId

b_id: GraphTensorId

c_id: GraphTensorId

MatMul

(B x M x K) * (B x K x N) = (B x M x N) out = out * alpha + beta * lhs * rhs

Fields

l_id: GraphTensorId

r_id: GraphTensorId

o_id: Option<GraphTensorId>

k: usize

alpha: T

beta: T

Rand

Fill with uniform random values in [0, 1).

Randn

Fill with normally distributed random values (mean, std).

Fields

mean: T

std: T

Permute

Fields

v_id: GraphTensorId

NoOp

Trait Implementations

impl<T: Clone + DType> Clone for Op<T>

fn clone(&self) -> Op<T>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<T: Debug + DType> Debug for Op<T>

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

Formats the value using the given formatter.

impl<T: PartialEq + DType> PartialEq for Op<T>

fn eq(&self, other: &Op<T>) -> 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<T: DType> StructuralPartialEq for Op<T>