def Cells

bendtorch-core@0.1.0.0 · src/tensor/parallel_proofs.bend:24

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

24# Every slot of `out`, the cells `base .. base + 2^depth`, holds what CellIs
25# names for its index.
26def Cells(~P: Data, ~A: Type, ~B: Type, ~cell: @+p: P -> @+position: Nat -> (A & B) -> Bool -> F32 & (A & B),
27  +p: P, -start: (A & B), +depth: Nat, +count: Nat, +base: Nat, -out: Array<F32>) -> Type:
28  @+path: Model.Path(depth) ->
29    CellIs(~P, ~A, ~B, ~cell, p, start, count, Nat.add(Model.offset(depth, path), base), Model.observe(F32, depth, out, path))