def node_joined

bendtorch-core@0.1.0.0 · src/tensor/parallel.bend:97

import Base
import bendtorch-core@0.1.0.0/src/storage/model.bend as Model

97def node_joined(~A: Type, ~B: Type, +inner: Nat, -rec: A -> B -> Array<F32> & (A & B), -low: Array<F32> & (A & B),
98  a: Array<F32>, -l: A, -r: B, same: {(a, (l, r)) == low : Array<F32> & (A & B)}, balanced: Model.Balanced(F32, inner, a),
99  high: Certified<(A & B), inner, rec(l, r)>) -> Certified<(A & B), 1n+inner, node(~A, ~B, rec, low)>:
100  Certified{result, high_same, high_balanced} = high
101  (b, rest) = result
102  Certified{(ANode{a, b}, rest),
103    %same : {(ANode{a, b}, rest) == node(~A, ~B, rec, _) : Array<F32> & (A & B)}
104    %high_same : {(ANode{a, b}, rest) == join(~(A & B), a, _) : Array<F32> & (A & B)}
105    {==},
106    (balanced, high_balanced)}