def lcols
bend-ml-tensor-array@0.1.19.0 · main.bend:122
import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray
import Base
122def lcols(mleft: Nat, +kn: Nat, +ia0: U32, +ib0: U32, +sa: U32, +sb: U32, +bcol: U32, out: List<&2, F32>, r: R) -> S3:
123 match mleft r:
124 case 0n R{a, b, +x}:
125 S3{a, b, x <> out}
126 case 1n+p R{a, b, +x}:
127 lcols(p, kn, ia0, (ib0 + bcol : U32), sa, sb, bcol, x <> out, dotg(kn, ia0, (ib0 + bcol : U32), sa, sb, 0.0, Array.get(F32, a, ia0), Array.get(F32, b, (ib0 + bcol : U32))))