Struct CodePredictorEngine 

pub struct CodePredictorEngine { /* private fields */ }

Implementations

impl CodePredictorEngine

pub fn open( store: &Qwen3TtsWeightStore, cp: &CodePredictorConfig, device: Device, ) -> Result<CodePredictorEngine, Error>

pub fn is_eager(&self) -> bool

pub fn talker_codec_flat(&self) -> (&[f32], usize)

Talker codec embedding table flat (row-major [codec_vocab, hidden]). Used by the speculative path to do cheap g0 group-embedding swaps when synthesising verifier inputs from drafted g0 proposals.

pub fn device(&self) -> Device

pub fn cp_backend_label(&self) -> String

pub fn warmup(&mut self, max_frames: usize) -> Result<(), Error>

pub fn predict_groups_slice( &mut self, talker_hidden: &[f32], group0: u32, ) -> Result<Vec<u32>, Error>

pub fn predict_groups_fill_emb( &mut self, talker_hidden: &[f32], group0: u32, pad: &[f32], codec_emb: &mut [f32], ) -> Result<Vec<u32>, Error>

CP predict + codec embed sum + pad (fused on eager).

pub fn predict_groups( &mut self, talker_hidden: ArrayBase<ViewRepr<&f32>, Dim<[usize; 1]>>, group0: u32, ) -> Result<Vec<u32>, Error>

pub fn sum_codec_groups_into( &self, groups: &[u32], out: &mut [f32], ) -> Result<(), Error>

Sum codec group embeddings into out (group 0 = talker table).

pub fn sum_codec_groups(&self, groups: &[u32]) -> Result<Vec<f32>, Error>

pub fn cp_step_embeds_from_groups( &self, groups: &[u32], ) -> Result<Vec<Vec<f32>>, Error>

Per-step codec embeds for fused backbone (groups[1..] → cp_step_embed_{i}).

pub fn codec_embed_row( &self, group_idx: usize, token: u32, ) -> Result<Vec<f32>, Error>