def gws

bend-ml-llama@0.2.0.0 · main.bend:809

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

import Base

809def gws(gs: List<&2, List<&2, Head>>, ks: List<&1, Array<F32>>, vs: List<&1, Array<F32>>) -> List<&1, GW>:
810  match gs ks vs:
811    case Con{g, gt} Con{k, kt} Con{v, vt}:
812      GW{g, k, v} <> gws(gt, kt, vt)
813    case _ _ _:
814      Nil{}

Uses (5)

Used by (1)

bend-ml-llama@0.2.0.0/main:attention