def treef

bendtorch@1.3.0.0 · PROOF.bend:525

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

525def treef(d: Nat, gs: T.Halves<T.FS>, +n: Nat, +h: Nat, +dm: T.Dm) -> MRF<d, gs, n, h, dm>:
526  match d gs:
527    case 0n T.Halves{l, r}:
528      fm0a(l, r, n, h, dm, lf1(l, h, dm, Nil{}))
529    case 1n++e T.Halves{l, r}:
530      fm1a(e, l, r, n, h, dm, cg(TA.half(TA.half(n)), l, Nil{}), cg(TA.half(Nat.sub(n, TA.half(n))), r, Nil{}), g1 => treef(e, g1, TA.half(n), h, dm), g2 => treef(e, g2, Nat.sub(n, TA.half(n)), h, dm))