def write_done

bendtorch-core@0.1.0.0 · src/tensor/from_values_contract.bend:198

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/LAWS.bend as StorageLaws
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/lifecycle.bend as Lifecycle
import bendtorch-core@0.1.0.0/src/tensor/word_conversion.bend as Conversion
import bendtorch-core@0.1.0.0/src/tensor/observation.bend as Observation

198def write_done(+shape: Shape.Shape, +prefix: List<&2, StorageLaws.Op>,
199  -valid: {Lifecycle.logical_length(base(shape, prefix)) == Shape.logical_numel(Shape.dimensions(shape)) : Nat}, +index: U32, +value: F32,
200  inside: {Nat.is_lt(U32.to_nat(index), U32.to_nat(Shape.count(shape))) == True{} : Bool}) ->
201  {Observation.write_done_pair(shape, Core.write_flat(shape, Core.TTensor{base(shape, prefix), valid}, index, value)) == True{} : Bool}:
202  Equal.trans(Bool, Observation.write_done_pair(shape, Core.write_flat(shape, Core.TTensor{base(shape, prefix), valid}, index, value)),
203    Result.is_done(&2, &2, Storage.Error, Unit, StorageLaws.status(base(shape, prefix), index, value)), True{},
204    done_projection(shape, base(shape, prefix), valid, index, value, Storage.set(base(shape, prefix), index, value), {==}),
205    Equal.cong(Result<&2, &2, Storage.Error, Unit>, Bool, Result.is_done(&2, &2, Storage.Error, Unit),
206      StorageLaws.status(base(shape, prefix), index, value), Done{Unit{}},
207      StorageLaws.replay_set(Shape.count(shape), 0.0, prefix, index, value, Conversion.shape_count_within(shape), inside)))