def node_split
bendtorch-core@0.1.0.0 · src/tensor/parallel.bend:108
import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel
import Base
import bendtorch-core@0.1.0.0/src/storage/model.bend as Model
108def node_split(~A: Type, ~B: Type, +inner: Nat, -rec: A -> B -> Array<F32> & (A & B), -low: Array<F32> & (A & B),
109 result: Array<F32> & (A & B), same: {result == low : Array<F32> & (A & B)},
110 balanced: Model.Balanced(F32, inner, Pair.fst(Array<F32>, A & B, result)),
111 next: @l: A -> @r: B -> Certified<(A & B), inner, rec(l, r)>) -> Certified<(A & B), 1n+inner, node(~A, ~B, rec, low)>:
112 (a, state) = result
113 (l, r) = state
114 node_joined(~A, ~B, inner, rec, low, a, l, r, same, balanced, next(l, r))