def parallel_allocated_retains

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

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/elementwise.bend as Elementwise
import bendtorch-core@0.1.0.0/src/tensor/elementwise_totality.bend as Totality
import bendtorch-core@0.1.0.0/src/tensor/elementwise_parallel_proof.bend as Parallel

438def parallel_allocated_retains(+operation: Elementwise.Operation, +plan: Shape.BroadcastPlan, left: Core.Dynamic, right: Core.Dynamic,
439  result: Result<&2, &1, Core.Error, Core.Tensor<Shape.output_shape(plan)>>) ->
440  Totality.Retention<left, right, Elementwise.parallel_allocated(operation, plan, left, right, result)>:
441  match result:
442    case Fail{error}:
443      Totality.Retention{property => {==}, property => {==}}
444    case Done{Core.TTensor{storage, exact}}:
445      match storage:
446        case Storage.Empty{}:
447          Totality.Retention{property => {==}, property => {==}}
448        case Storage.Owned{+depth, +length, array, balanced, valid}:
449          output_retains(operation, plan, depth, length, valid, exact, left, right, Elementwise.take(left), Elementwise.take(right),
450            Parallel.taken(left), Parallel.taken(right))