def written_done

bendtorch-core@0.1.0.0 · src/tensor/reshape_contract.bend:156

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/read_contract.bend as Reads
import bendtorch-core@0.1.0.0/src/tensor/transform_observation.bend as Observation
import bendtorch-core@0.1.0.0/src/tensor/write_contract.bend as Writes

156def written_done(+input: Shape.Shape, +fill: F32, +target: Shape.Shape, -valid: {Lifecycle.logical_length(StorageLaws.created(Shape.count(input), fill)) == Shape.logical_numel(Shape.dimensions(input)) : Nat}, +index: U32, +value: F32, +other: U32, +same: {Shape.logical_numel(Shape.dimensions(input)) == Shape.logical_numel(Shape.dimensions(target)) : Nat}, inside: {Nat.is_lt(U32.to_nat(other), Shape.logical_numel(Shape.dimensions(target))) == True{} : Bool}) ->
157  {Observation.reshape_written(input, index, value, Shape.dimensions(target), other, Done{Core.TTensor{StorageLaws.created(Shape.count(input), fill), valid}}) ==
158    Some{Writes.written_pick(input, index, value, other, fill)} : Maybe<&2, F32>}:
159  Equal.trans(Maybe<&2, F32>,
160    Observation.reshape_written_pair(input, Shape.dimensions(target), other,
161      Core.write_result(input, StorageLaws.created(Shape.count(input), fill), valid, index, value, Storage.set(StorageLaws.created(Shape.count(input), fill), index, value), {==})),
162    Lifecycle.read_value(Storage.get(StorageLaws.written(StorageLaws.created(Shape.count(input), fill), index, value), other)),
163    Some{Writes.written_pick(input, index, value, other, fill)},
164    written_projection(input, target, StorageLaws.created(Shape.count(input), fill), valid, index, value, other, same, Storage.set(StorageLaws.created(Shape.count(input), fill), index, value), {==}),
165    Reads.stored_value(StorageLaws.written(StorageLaws.created(Shape.count(input), fill), index, value), other, Writes.written_pick(input, index, value, other, fill),
166      StorageLaws.replay_get(Shape.count(input), fill, [StorageLaws.Write{index, value}], other, Conversion.shape_count_within(input),
167        Reads.word_inside(input, other, inside_input(input, target, other, same, inside)))))