def planned_retains

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

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
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

452def planned_retains(+operation: Elementwise.Operation, left: Core.Dynamic, right: Core.Dynamic,
453  result: Result<&2, &2, Shape.Error, Shape.BroadcastPlan>) ->
454  Totality.Retention<left, right, Elementwise.parallel_planned(operation, left, right, result)>:
455  match result:
456    case Fail{error}:
457      Totality.Retention{property => {==}, property => {==}}
458    case Done{+plan}:
459      parallel_allocated_retains(operation, plan, left, right, Core.new(Shape.output_shape(plan), 0.0))