def update_checked_model

bendtorch-core@0.1.0.0 · src/tensor/core_PROOF.bend:305

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/storage/main.bend as Storage
import bendtorch-core@0.1.0.0/src/storage/model.bend as Model
import bendtorch-core@0.1.0.0/src/storage/refinement.bend as Refinement
import bendtorch-core@0.1.0.0/src/storage/route.bend as Route
import bendtorch-core@0.1.0.0/src/storage/LAWS.bend as StorageLaws
import bendtorch-core@0.1.0.0/src/tensor/core_LAWS.bend as Laws

305def update_checked_model(-shape: Shape.Shape, +depth: Nat, +length: Nat, -array: Array<F32>, +index: U32, +value: F32,
306  allowed: Bool, -decision: {allowed == Nat.is_lt(U32.to_nat(index), length) : Bool},
307  -balanced: Model.Balanced(F32, depth, array), -valid: {Storage.bounds(depth, length) == True{} : Bool},
308  -length_exact: {length == Shape.logical_numel(Shape.dimensions(shape)) : Nat},
309  -same: {Storage.set_checked(depth, length, array, index, value, allowed, decision, balanced, valid) ==
310    Storage.set(Storage.Owned{depth, length, array, balanced, valid}, index, value) : Storage.Storage & Result<&2, &2, Storage.Error, Unit>}) ->
311  UpdateCheckedModel<shape, depth, length, array, index, value, allowed, decision, balanced, valid, length_exact, same>:
312  match allowed:
313    case False{}:
314      UpdateCheckedModel{Unit{}}
315    case True{}:
316      UpdateCheckedModel{(Route.index_matched(depth, length, index, valid, Equal.sym(Bool, True{}, Nat.is_lt(U32.to_nat(index), length), decision)),
317        (StorageLaws.primitive_write_read(F32, depth, array, index, value, balanced),
318          other => checked => different => (checked,
319            %Laws.checked_route_separation(depth, length, index, other,
320              Route.index_matched(depth, length, index, valid, Equal.sym(Bool, True{}, Nat.is_lt(U32.to_nat(index), length), decision)),
321              checked, different) :
322              {Model.observe(F32, depth, Array.set(F32, array, index, value), Refinement.route(depth, other)) ==
323                Model.select(F32, _, value, Model.observe(F32, depth, array, Refinement.route(depth, other))) : Maybe<&2, F32>}
324            StorageLaws.primitive_update_frame(F32, depth, array, index, Refinement.route(depth, other), value, balanced))))}