Function tp_decode_step 

pub fn tp_decode_step(
    spec: &ModelSpec,
    dir: &str,
    ctx: usize,
    tok: usize,
    pos: usize,
    tp: usize,
) -> Result<TpAuthored, NotYet>

A whole decode step over tp chips: the embedding lookup and the head replicated, every layer split.