def head_att

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

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

import Base

785def head_att(+hd: Nat, +t: Nat, +scale: F32, k: Array<F32>, v: Array<F32>, +q: List<&2, F32>) -> HO:
786  head_sr(hd, t, scale, v, scores(hd, t, k, q))

Uses (7)

Used by (1)

bend-ml-llama@0.2.1.0/main:group_att