def b1_g

bendtorch-gpu@1.3.0.0 · main.bend:811

import Base
import bendtorch@1.3.0.0/main.bend as T

811def b1_g(+e: U32, q: Array<F32>, k: Array<F32>, p: Array<F32>, g: Array<F32>, r: T.D2) -> S6:
812  match r:
813    case T.D2{d, v, +s}: S6{q, k, v, d, p, Array.set(F32, g, e, s)}