def kv_of

bendtorch@1.3.0.0 · main.bend:2353

import Base

2353# the key/value head that query head hi reads, with g query heads per key/value head; the forward and the backward pass
2354# both pick it here (the law kv_of_group says which heads share one)
2355def kv_of(+hi: Nat, +g: Nat) -> Nat:
2356  Nat.div(hi, g)