def Out.then

bend-ml-llama@0.2.1.0 · main.bend:1012

import Base

1012# continues with the fields of a position's result: logits, final norm, activations, caches, model
1013def Out.then(-R: Type, -c: Cfg, -cap: Nat, -t: Nat, o: Out<c, cap, t>, k: List<&2, F32> -> List<&2, F32> -> List<&2, Act> -> List<&1, KV<cap, 1n+t>> -> Model<c> -> R) -> R:
1014  match o:
1015    case Out{lg, nf, acts, cs, m}:
1016      k(lg, nf, acts, cs, m)