def set_checked_length

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

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

47def set_checked_length(+d: Nat, +n: Nat, array: Array<F32>, +i: U32, +value: F32, allowed: Bool,
48  -decision: {allowed == Nat.is_lt(U32.to_nat(i), n) : Bool},
49  -balanced: Model.Balanced(F32, d, array), -valid: {Storage.bounds(d, n) == True{} : Bool}) ->
50  {retained_write(Storage.set_checked(d, n, array, i, value, allowed, decision, balanced, valid)) == n : Nat}:
51  match allowed:
52    case False{}:
53      {==}
54    case True{}:
55      {==}