def dynamic_value

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

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/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/lifecycle.bend as Lifecycle

80def dynamic_value(+output: Shape.Shape, -storage: Storage.Storage,
81  -fits: {Lifecycle.logical_length(storage) == Shape.logical_numel(Shape.dimensions(output)) : Nat}, +other: U32,
82  pair: Storage.Storage & Result<&2, &2, Storage.Error, F32>, -equal: {pair == Storage.get(storage, other) : Storage.Storage & Result<&2, &2, Storage.Error, F32>}) ->
83  {Core.dynamic_read_value(Core.read_dynamic_result(output, Core.read_result(output, storage, fits, other, pair, equal))) ==
84    Lifecycle.read_value(pair) : Maybe<&2, F32>}:
85  (kept, result) = pair
86  match result:
87    case Fail{error}:
88      {==}
89    case Done{value}:
90      {==}