def t_layer

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

import bend-ml-llama@0.2.1.0/main.bend as BendMlLlama

import Base

1626# the tensors in w/st.txt's order (reference/prep.py's weights()): emb, then layer i's ln1 q k v o ln2 gate up down
1627# (j = 0..8), then lnf
1628def t_layer(+i: Nat, +j: Nat) -> Nat:
1629  Nat.add(Nat.add(1n, Nat.mul(9n, i)), j)

Uses (3)

Used by (2)