def output_retains

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

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

426def output_retains(+operation: Elementwise.Operation, +plan: Shape.BroadcastPlan, +depth: Nat, +length: Nat,
427  -valid: {Storage.bounds(depth, length) == True{} : Bool},
428  -exact: {length == Shape.logical_numel(Shape.dimensions(Shape.output_shape(plan))) : Nat},
429  -left: Core.Dynamic, -right: Core.Dynamic, taken_left: Core.Dynamic & Array<F32>, taken_right: Core.Dynamic & Array<F32>,
430  -left_taken: Parallel.Taken<left, taken_left>, -right_taken: Parallel.Taken<right, taken_right>) ->
431  Totality.Retention<left, right, Elementwise.parallel_output(operation, plan, depth, length, valid, exact, taken_left, taken_right)>:
432  (kept_left, la) = taken_left
433  (kept_right, ra) = taken_right
434  Totality.Retention{
435    {\{Parallel.Taken: depth => balanced => reads => same => same} : Parallel.Taken<left, (kept_left, la)> -> Totality.SameSource(left, kept_left)}(left_taken),
436    {\{Parallel.Taken: depth => balanced => reads => same => same} : Parallel.Taken<right, (kept_right, ra)> -> Totality.SameSource(right, kept_right)}(right_taken)}