def expand
import bend-collections@1.0.0.0/src/crypto/aes/aes.bend as Aesimport Base
import bend-collections@1.0.0.0/src/crypto/aes/types.bend as T
145def expand(+nk: Nat, +nr: Nat, key: List<&2, U32>) -> List<&2, T.State>:
146 round_keys(List.reverse(&2, T.Quad, grow(Nat.sub(Nat.mul(4n, 1n+nr), nk), nk, nk,
147 List.reverse(&2, T.Quad, key_words(key)))))