def loop_after_pair

bendtorch-core@0.1.0.0 · src/tensor/transform_PROOF.bend:913

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout
import bendtorch-core@0.1.0.0/src/tensor/transform_LAWS.bend as Laws
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/transforms.bend as Transforms

913def loop_after_pair(+fuel: Nat, +input: Shape.Shape, +axes: List<&2, U32>, +plan: Layout.Plan<input, axes>, +cursor: Nat,
914  -original: Core.Tensor<input>, -before_source: Core.Tensor<input>, -before_target: Core.Tensor<Layout.output(input, axes, plan)>,
915  -index: U32, -converted: {U32.to_nat(index) == cursor : Nat}, -candidate: {index == U32.from_nat(cursor) : U32},
916  -prefix: Transforms.Prefix<input, axes, plan, original, before_source, before_target, cursor>,
917  pair: Core.Tensor<input> & Result<&2, &1, Transforms.Error, Core.Tensor<Layout.output(input, axes, plan)>>,
918  natural_exact: {Transforms.NaturalCopied{pair, index, converted, candidate} ==
919    Transforms.natural_step(input, axes, plan, cursor, before_source, before_target) : Transforms.NaturalCopy<input, axes, plan, cursor>},
920  -meaning: Transforms.StepMeaning(input, axes, plan, index, before_source, before_target, pair),
921  rec: @source: Core.Tensor<input> -> @target: Core.Tensor<Layout.output(input, axes, plan)> ->
922    @-prefix: Transforms.Prefix<input, axes, plan, original, source, target, 1n+cursor> ->
923      Transforms.LoopCertificate<input, axes, plan, original, fuel, 1n+cursor, source, target>) ->
924  Transforms.LoopCertificate<input, axes, plan, original, 1n+fuel, cursor, before_source, before_target>:
925  (source, result) = pair
926  match result:
927    case Fail{error}:
928      Transforms.LoopCertificate{(source, Fail{error}),
929        Equal.cong(Transforms.NaturalCopy<input, axes, plan, cursor>,
930          Core.Tensor<input> & Result<&2, &1, Transforms.Error, Core.Tensor<Layout.output(input, axes, plan)>>,
931          step => Transforms.natural_next(input, axes, plan, cursor,
932            source => target => Transforms.copy_loop(fuel, input, axes, plan, source, target, 1n+cursor), step),
933          Transforms.NaturalCopied{(source, Fail{error}), index, converted, candidate},
934          Transforms.natural_step(input, axes, plan, cursor, before_source, before_target), natural_exact), Unit{}}
935    case Done{target}:
936      loop_continue(fuel, input, axes, plan, cursor, original, before_source, before_target, source, target, index, converted, candidate,
937        natural_exact, rec(source)(target)(
938          Laws.advance_prefix(input, axes, plan, original, before_source, source, before_target, target, cursor, prefix,
939            actual_advanced(input, axes, plan, cursor, original, before_source, source, before_target, target, prefix,
940              canonical_step(input, axes, plan, cursor, index, before_source, before_target, source, target, candidate, converted, meaning)))))