~/jendhub

def step_indices_certificate

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

import bendtorch-core@0.1.0.0/src/tensor/core_PROOF.bend as CorePROOF
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

609def step_indices_certificate(+operation: Elementwise.Operation, +plan: Shape.BroadcastPlan, +index: U32,
610  state: Elementwise.State<Shape.output_shape(plan)>, result: Result<&2, &1, Core.Error, U32 & U32>,
611  exact: {result == Elementwise.input_indices(plan, index) : Result<&2, &1, Core.Error, U32 & U32>}) -> Elementwise.StepCertificate<operation, plan, index, state>:
612  Elementwise.State{left, right, output} = state
613  match result:
614    case Fail{error}:
615      Elementwise.StepCertificate{Elementwise.Stop{left, right, error},
616        Equal.cong(Result<&2, &1, Core.Error, U32 & U32>, Elementwise.Step<Shape.output_shape(plan)>,
617          input => Elementwise.step_indices(operation, Shape.output_shape(plan), index, Elementwise.State{left, right, output}, input),
618          Fail{error}, Elementwise.input_indices(plan, index), exact), Unit{}}
619    case Done{indices}:
620      (+left_index, +right_index) = indices
621      step_left_certificate(operation, plan, index, left, right, output, left_index, right_index, exact,
622        result => equality => Equal.trans(Elementwise.Step<Shape.output_shape(plan)>, result,
623          Elementwise.step_indices(operation, Shape.output_shape(plan), index, Elementwise.State{left, right, output}, Done{(left_index, right_index)}),
624          Elementwise.step(operation, plan, index, Elementwise.State{left, right, output}), equality,
625          Equal.cong(Result<&2, &1, Core.Error, U32 & U32>, Elementwise.Step<Shape.output_shape(plan)>,
626            input => Elementwise.step_indices(operation, Shape.output_shape(plan), index, Elementwise.State{left, right, output}, input),
627            Done{(left_index, right_index)}, Elementwise.input_indices(plan, index), exact)),
628        Core.dynamic_read_certificate(left, left_index))

Uses (19)

Used by (1)