def sl_rows

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

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

import Base

1274# rows i.. of a slice: w numbers a row from src[i*st + o], into d[i*w]
1275def sl_rows(r: Nat, +i: U32, +w: Nat, +w32: U32, +st: U32, +o: U32, sl: SL) -> SL:
1276  match r sl:
1277    case 0n sl: sl
1278    case 1n+p SL{a, d}: sl_rows(p, (i + 1 : U32), w, w32, st, o, sl_row(w, ((i * st : U32) + o : U32), (i * w32 : U32), d, Array.get(F32, a, ((i * st : U32) + o : U32))))

Uses (8)

Used by (2)