Skip to main content

vae_resnet

Function vae_resnet 

Source
pub fn vae_resnet(a: &VaeResnetArgs<'_>, x: &[f32], out: &mut [f32]) -> bool
Expand description

One whole VAE resnet block on the device (norm+silu → conv ×2 → shortcut → add, one command buffer).