def Flat.rowsum.go

bend-tensors@0.0.0.2 · bend_tensors.bend:1998

import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
import Base

1998def Flat.rowsum.go(n: Nat, +i: U32, r: Array<F32> & F32, +acc: F32) -> Array<F32> & F32:
1999  match n:
2000    case 0n:
2001      (a, x) = r
2002      (a, (acc + x : F32))
2003    case 1n+p:
2004      (a, x) = r
2005      Flat.rowsum.go(p, (i + 1 : U32), Array.get(F32, a, (i + 1 : U32)), (acc + x : F32))

Uses (9)

Used by (1)