Function tp_stacked_layer 

pub fn tp_stacked_layer(
    spec: &ModelSpec,
    l: usize,
    dir: &str,
    ctx: usize,
    poss: &[usize],
    tp: usize,
) -> Result<TpAuthored, NotYet>

Layer l of a batched tensor-parallel decode step alone.