Enum Op 

pub enum Op<D>
where
    D: Device,
{
    Unary(UnaryOp, LazyBuffer<D>),
    Binary(BinaryOp, LazyBuffer<D>, LazyBuffer<D>),
    MatMul(LazyBuffer<D>, LazyBuffer<D>, (usize, usize, usize, usize), bool),
    Reduce(ReduceOp, LazyBuffer<D>, usize),
    Const(Const),
    Value,
    Layout(LayoutOp, LazyBuffer<D>),
    Reshape(LazyBuffer<D>),
    Custom {
        f: CustomF<<D as Device>::Slice>,
        args: Vec<LazyBuffer<D>>,
    },
    CustomIp {
        f: CustomF<<D as Device>::Slice>,
        args: Vec<LazyBuffer<D>>,
        src: LazyBuffer<D>,
    },
    Ssa {
        ssa: Kernel,
        args: Vec<LazyBuffer<D>>,
    },
    Set {
        values: LazyBuffer<D>,
        src: LazyBuffer<D>,
        dst_layout: Layout,
    },
}

Variants

Unary(UnaryOp, LazyBuffer<D>)

Binary(BinaryOp, LazyBuffer<D>, LazyBuffer<D>)

MatMul(LazyBuffer<D>, LazyBuffer<D>, (usize, usize, usize, usize), bool)

Reduce(ReduceOp, LazyBuffer<D>, usize)

Const(Const)

Value

Values are leafs of the op DAG, holding values that do not depend on other values.

Layout(LayoutOp, LazyBuffer<D>)

Reshape(LazyBuffer<D>)

Custom

Fields

f: CustomF<<D as Device>::Slice>

args: Vec<LazyBuffer<D>>

CustomIp

In-place variant of the custom op.

Fields

f: CustomF<<D as Device>::Slice>

args: Vec<LazyBuffer<D>>

src: LazyBuffer<D>

Ssa

Custom op based on some SSA code.

Fields

ssa: Kernel

args: Vec<LazyBuffer<D>>

Set

Set the data from src using data from values. The LazyBuffer is shared between self and src.

Fields

values: LazyBuffer<D>

src: LazyBuffer<D>

dst_layout: Layout

Trait Implementations

impl<D> Debug for Op<D>

where D: Debug + Device, <D as Device>::Slice: Debug,

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

Formats the value using the given formatter.