Enum PvStep 

pub enum PvStep {
    Conv {
        wi: usize,
        src: u8,
        dst: u8,
        res: Option<u8>,
        dil: u32,
        h: usize,
        w: usize,
    },
    NormSilu {
        gi: usize,
        src: u8,
        dst: u8,
        c: usize,
        h: usize,
        w: usize,
    },
    Unshuffle {
        src: u8,
        dst: u8,
        c: usize,
        h: usize,
        w: usize,
    },
    Shuffle {
        src: u8,
        dst: u8,
        add: Option<u8>,
        c: usize,
        h: usize,
        w: usize,
    },
}

One device step. Buffers are numbered; h×w is the spatial size of the step’s SOURCE, NCHW f32 throughout.

Variants

Conv

dst = conv(src) + b (+ res), 3×3, dilation = padding = dil.

Fields

wi: usize

src: u8

dst: u8

res: Option<u8>

dil: u32

h: usize

w: usize

NormSilu

dst = silu(rmsnorm_c(src) · γ).

Fields

gi: usize

src: u8

dst: u8

c: usize

h: usize

w: usize

Unshuffle

dst = pixel_unshuffle(src, 2): [c, h, w] → [4c, h/2, w/2].

Fields

src: u8

dst: u8

c: usize

h: usize

w: usize

Shuffle

dst = pixel_shuffle(src, 2) (+ add): [c, h, w] → [c/4, 2h, 2w].

Fields

src: u8

dst: u8

add: Option<u8>

c: usize

h: usize

w: usize

Trait Implementations

impl Clone for PvStep

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for PvStep

impl Debug for PvStep

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

Formats the value using the given formatter.