Enum ComputeShape 

pub enum ComputeShape<'a> {
    Tiles {
        block: &'a BlockShape,
        sub_blocks: (usize, usize),
        operand: Precision,
    },
    AccMap {
        body: TensorBody,
        reads: Vec<AccRead>,
    },
    Generic {
        iters: &'a [IterSpec],
        ins: &'a [ElemAccess],
        out: &'a ElemAccess,
        body: &'a TensorBody,
    },
    Scalar(&'a ScalarBody),
}

A ComputeOp as the shapes a printer that computes in software tells apart: a tile product, an elementwise map over the acquired accumulator, the general form, a scalar body. Such a printer matches on what an op computes and never on which epilogue it is. Requant, bias and ReLU are each an AccMap, whose body it prints as it prints a Generic’s.

Variants

Tiles

Fields

block: &'a BlockShape

sub_blocks: (usize, usize)

operand: Precision

AccMap

Element e of every accumulator tile k becomes body, its operand i the stage’s ins[i] read as reads[i] says. Its arithmetic wraps at 64 bits, as the epilogues it stands for do.

Fields

body: TensorBody

reads: Vec<AccRead>

Generic

Fields

iters: &'a [IterSpec]

ins: &'a [ElemAccess]

out: &'a ElemAccess

body: &'a TensorBody

Scalar(&'a ScalarBody)

Trait Implementations

impl<'a> Clone for ComputeShape<'a>

fn clone(&self) -> ComputeShape<'a>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for ComputeShape<'a>

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

Formats the value using the given formatter.