def read_checked_keeps

bendtorch-core@0.1.0.0 · src/tensor/lifecycle.bend:258

import Base
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

258def read_checked_keeps(+depth: Nat, +length: Nat, -array: Array<F32>, +index: U32, allowed: Bool,
259  -decision: {allowed == Nat.is_lt(U32.to_nat(index), length) : Bool},
260  -balanced: Model.Balanced(F32, depth, array), -valid: {Storage.bounds(depth, length) == True{} : Bool}) ->
261  ReadCheckedKeeps<depth, length, array, index, allowed, decision, balanced, valid>:
262  match allowed:
263    case False{}:
264      ReadCheckedKeeps{other => {==}}
265    case True{}:
266      ReadCheckedKeeps{other => Equal.trans(Maybe<&2, F32>,
267        retained_observe(Storage.finish_read(depth, length, array, index, Array.get(F32, array, index), {==}, balanced, valid), other),
268        checked_observe_array(depth, length, Refinement.first(F32, Array.get(F32, array, index)), other),
269        checked_observe_array(depth, length, array, other),
270        finish_read_observe(depth, length, array, index, other, Array.get(F32, array, index), {==}, balanced, valid),
271        Equal.cong(Array<F32>, Maybe<&2, F32>, a => checked_observe_array(depth, length, a, other),
272          Refinement.first(F32, Array.get(F32, array, index)), array, Refinement.law_get_retains(F32, depth, array, index, balanced)))}