Struct Qi21VaeDecodeArgs 

pub struct Qi21VaeDecodeArgs<'a> {
    pub key: u64,
    pub post_quant: Qi21VaeConvRef<'a>,
    pub conv_in: Qi21VaeConvRef<'a>,
    pub mid_res: [Qi21VaeResRef<'a>; 2],
    pub attn_gamma: &'a [f32],
    pub attn_qkv: Qi21VaeConvRef<'a>,
    pub attn_proj: Qi21VaeConvRef<'a>,
    pub ups: Vec<Qi21VaeUpRef<'a>>,
    pub norm_out: &'a [f32],
    pub conv_out: Qi21VaeConvRef<'a>,
}

The Qwen-Image-2.1 VAE decoder (qwen_image21_vae.rs) for a resident device decode.

Fields

key: u64

Identity of the weights: a backend caches its planes by it.

post_quant: Qi21VaeConvRef<'a>

conv_in: Qi21VaeConvRef<'a>

mid_res: [Qi21VaeResRef<'a>; 2]

attn_gamma: &'a [f32]

The mid attention: RMS_norm γ, to_qkv (1×1, 3c outputs: q, k, v), proj (1×1); single head, residual.

attn_qkv: Qi21VaeConvRef<'a>

attn_proj: Qi21VaeConvRef<'a>

ups: Vec<Qi21VaeUpRef<'a>>

norm_out: &'a [f32]

conv_out: Qi21VaeConvRef<'a>