Function dsa_tensor_specs 

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

Every self_attn tensor a DSA block has — and the complete list of what it may have. full_heads is the pre-shard head count; the checkpoint is never sharded on disk, so binding always validates against the full geometry and slices after.