def fpar_full

bendtorch@1.3.0.0 · main.bend:2973

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

2973def fpar_full(d: Nat, gs: Halves<FS>, +n: Nat, +h: Nat, +dm: Dm) -> List<&1, Array<F32>>:
2974  match d gs:
2975    case 0n Halves{l, r}:
2976      a b = fleaf_full(l, h, dm, Nil{}) fleaf_full(r, h, dm, Nil{})
2977      List.append(&1, Array<F32>, a, b)
2978    case 1n++e Halves{l, r}:
2979      a b = fpar_full(e, cut(FS, TA.half(TA.half(n)), l, Nil{}), TA.half(n), h, dm) fpar_full(e, cut(FS, TA.half(Nat.sub(n, TA.half(n))), r, Nil{}), Nat.sub(n, TA.half(n)), h, dm)
2980      List.append(&1, Array<F32>, a, b)