def write_rows

bend-ml-llama@0.2.1.0 · main.bend:491

import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

491# writes row j of rs into array j, at off
492def write_rows(arrs: List<&1, Array<F32>>, rs: List<&2, List<&2, F32>>, +off: U32) -> List<&1, Array<F32>>:
493  match arrs rs:
494    case Con{a, at} Con{r, rt}:
495      TA.fill_list(r, off, a) <> write_rows(at, rt, off)
496    case _ _:
497      arrs