def build_balanced
bendtorch-core@0.1.0.0 · src/tensor/batched_matmul.bend:293
import bendtorch-core@0.1.0.0/src/tensor/batched_matmul.bend as BatchedMatmul
import Base
import bendtorch-core@0.1.0.0/src/storage/model.bend as Model
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise
import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel
293def build_balanced(+depth: Nat, +splits: Nat, +given: Layout, +count: Nat, +base: Nat, la: Array<F32>, ra: Array<F32>) ->
294 Model.Balanced(F32, depth, build(depth, splits, given, count, base, la, ra)):
295 Parallel.balanced(~Layout, ~Layout, ~Array<F32>, ~Array<F32>, ~layout, ~Elementwise.split, ~cell_at, given, depth, splits, count, base, (la, ra))