dsa_tensor_specs

metrale_model_arch::glm5next_dsa::binding

Function dsa_tensor_specs 

Source 
pub fn dsa_tensor_specs(
    cfg: &Glm5NextDsaConfig,
    full_heads: usize,
) -> Vec<DsaSpec>
Expand description

2026-09-25: Every self_attn tensor a DSA block has, and the only ones it may have. full_heads is the unsharded head count: shapes are checked against the full tensors, which build::build_dsa_weights shards afterwards.