def fork_balanced
bendtorch-core@0.1.0.0 · src/tensor/parallel.bend:157
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
157def fork_balanced(~A: Type, ~B: Type, +inner: Nat, pair: (A & B) & (A & B), -low: A -> B -> Array<F32>, -high: A -> B -> Array<F32>,
158 low_valid: @l: A -> @r: B -> Model.Balanced(F32, inner, low(l, r)),
159 high_valid: @l: A -> @r: B -> Model.Balanced(F32, inner, high(l, r))) ->
160 Model.Balanced(F32, 1n+inner, fork(~A, ~B, pair, low, high)):
161 (first, second) = pair
162 (l1, r1) = first
163 (l2, r2) = second
164 (low_valid(l1, r1), high_valid(l2, r2))