Skip to main content

qi21_vae_decode

cortiq_engine::gpu

Function qi21_vae_decode 

Source 
pub fn qi21_vae_decode(
    a: &Qi21VaeDecodeArgs<'_>,
    z: &[f32],
    h: usize,
    w: usize,
    out: &mut [f32],
) -> bool
Expand description

Resident VAE decode: z = raw latent planes [z_dim, h·w] → out [out_channels, 16h·16w] (before the clamp). false = not handled, out untouched: the per-conv path runs.