GatherPROOF.build_cells: def of bendtorch-core@0.1.0.0 Import: import bendtorch-core@0.1.0.0/src/tensor/gather_PROOF.bend as GatherPROOF Lines 336-353 of bendtorch-core@0.1.0.0/src/tensor/gather_PROOF.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather_PROOF.bend&start_line=336&end_line=353 import Base import bendtorch-core@0.1.0.0/src/storage/model.bend as Model import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gather import bendtorch-core@0.1.0.0/src/tensor/parallel_proofs.bend as Generic # Every slot of the forward build holds what its cell computes from (A, I). # The Fact holds the cells eta-expanded, as slice_PROOF.bend's build_cells. def build_cells(+depth: Nat, +splits: Nat, +plan: Gather.Plan, +cells: Nat, +base: Nat, +da: Nat, -A: Array, a: Array, -same: {a == A : Array}, -valid: Model.Balanced(F32, da, A), +di: Nat, -I: Array, index: Array, -same_index: {index == I : Array}, -index_valid: Model.Balanced(U32, di, I)) -> Generic.Fact, ~Array, ~Gather.cell, plan, (A, I), depth, cells, base, Gather.build(depth, splits, plan, cells, base, a, index))>: Generic.Fact{path => {\{Generic.Fact: q => q} : Generic.Fact, ~Array, ~Gather.cell, plan, (A, I), depth, cells, base, Gather.build(depth, splits, plan, cells, base, a, index))> -> Generic.Cells(~Gather.Plan, ~Array, ~Array, ~Gather.cell, plan, (A, I), depth, cells, base, Gather.build(depth, splits, plan, cells, base, a, index))}( Generic.build_cells(~Gather.Plan, ~Gather.Plan, ~Array, ~Array, ~Gather.same, ~Gather.split, ~Gather.cell, plan, (A, I), depth, splits, cells, base, (a, index), both(a, index, A, I, same, same_index), at => inside => {\{Generic.Fact: q => q} : Generic.Fact<{Pair.snd(F32, Array & Array, Gather.cell(plan, at, (A, I), inside)) == (A, I) : Array & Array}> -> {Pair.snd(F32, Array & Array, Gather.cell(plan, at, (A, I), inside)) == (A, I) : Array & Array}}( cell_keeps(plan, at, da, A, di, I, inside, valid, index_valid)), Pair.fst({Pair.fst(Array & Array, Array & Array, Gather.split((A, I))) == (A, I) : Array & Array}, {Pair.snd(Array & Array, Array & Array, Gather.split((A, I))) == (A, I) : Array & Array}, {\{Generic.Fact: q => q} : Generic.Fact -> SplitFacts(A, I)}(split_facts(da, A, di, I, valid, index_valid))), Pair.snd({Pair.fst(Array & Array, Array & Array, Gather.split((A, I))) == (A, I) : Array & Array}, {Pair.snd(Array & Array, Array & Array, Gather.split((A, I))) == (A, I) : Array & Array}, {\{Generic.Fact: q => q} : Generic.Fact -> SplitFacts(A, I)}(split_facts(da, A, di, I, valid, index_valid)))))(path)} Uses (21): https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Array https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32 https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Pair https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Pair.fst https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Pair.snd https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Sigma https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32 https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/storage/model.bend&name=Balanced https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather.bend&name=Plan https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather.bend&name=build https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather.bend&name=cell https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather.bend&name=same https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather.bend&name=split https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather_PROOF.bend&name=SplitFacts https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather_PROOF.bend&name=both https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather_PROOF.bend&name=cell_keeps https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/gather_PROOF.bend&name=split_facts https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel_proofs.bend&name=Cells https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel_proofs.bend&name=Fact https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel_proofs.bend&name=build_cells