def lrows

bend-ml-tensor-array@0.1.19.0 · main.bend:129

import Base

129def lrows(nleft: Nat, +mm1: Nat, +kn: Nat, +i: U32, +arow: U32, +sa: U32, +sb: U32, +bcol: U32, st: S3) -> S3:
130  match nleft st:
131    case 0n S3{a, b, out}:
132      S3{a, b, out}
133    case 1n+p S3{a, b, out}:
134      lrows(p, mm1, kn, (i + 1 : U32), arow, sa, sb, bcol, lcols(mm1, kn, (i * arow : U32), 0, sa, sb, bcol, out, dotg(kn, (i * arow : U32), 0, sa, sb, 0.0, Array.get(F32, a, (i * arow : U32)), Array.get(F32, b, 0))))