def write_result

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

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/tensor/lifecycle.bend as Lifecycle

235def write_result(-shape: Shape.Shape, -original: Storage.Storage,
236  -valid: {Lifecycle.logical_length(original) == Shape.logical_numel(Shape.dimensions(shape)) : Nat}, +i: U32, +value: F32,
237  pair: Storage.Storage & Result<&2, &2, Storage.Error, Unit>,
238  -same: {pair == Storage.set(original, i, value) : Storage.Storage & Result<&2, &2, Storage.Error, Unit>}) ->
239  Tensor<shape> & Result<&2, &2, Error, Unit>:
240  (storage, result) = pair
241  tensor : Tensor<shape> = TTensor{storage, Equal.trans(Nat, Lifecycle.logical_length(storage), Lifecycle.logical_length(original),
242    Shape.logical_numel(Shape.dimensions(shape)),
243    %Equal.sym(Storage.Storage & Result<&2, &2, Storage.Error, Unit>, (storage, result), Storage.set(original, i, value), same) :
244      {Lifecycle.retained_write(_) == Lifecycle.logical_length(original) : Nat}
245    Lifecycle.set_length(original, i, value), valid)}
246  (tensor, storage_write_error(result))