expected_replicated_shape

Function expected_replicated_shape 

Source
pub fn expected_replicated_shape(
    name: &str,
    config: &ModelConfig,
) -> Result<Vec<usize>>
Expand description

Storage geometry for weights that remain full-width on every TP rank. These are the matrix/vector dimensions consumed by cpu_bind::assemble. Sharded projections are validated by tp::plan_tensor_bytes instead.