def gws

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

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

import Base

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

Uses (5)

Used by (1)

bend-ml-llama@0.2.1.0/main:attention