def stepped

bendtorch-core@0.1.0.0 · src/tensor/gather.bend:308

import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gather

import Base

308# One gradient cell (o, k, r): its word is added when its entry is i.
309def stepped(+i: Nat, +at: Nat, +slot: Nat, acc: F32, state: Array<F32> & Array<U32>) -> F32 & (Array<F32> & Array<U32>):
310  (g, index) = state
311  compared(i, at, acc, g, Array.get(U32, index, U32.from_nat(slot)))

Uses (9)

Used by (3)