def loop_continue

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

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/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/transforms.bend as Transforms

891def loop_continue(+fuel: Nat, +input: Shape.Shape, +axes: List<&2, U32>, +plan: Layout.Plan<input, axes>, +cursor: Nat,
892  -original: Core.Tensor<input>, -before_source: Core.Tensor<input>, -before_target: Core.Tensor<Layout.output(input, axes, plan)>,
893  -after_source: Core.Tensor<input>, -after_target: Core.Tensor<Layout.output(input, axes, plan)>, -index: U32,
894  -converted: {U32.to_nat(index) == cursor : Nat}, -candidate: {index == U32.from_nat(cursor) : U32},
895  natural_exact: {Transforms.NaturalCopied{(after_source, Done{after_target}), index, converted, candidate} ==
896    Transforms.natural_step(input, axes, plan, cursor, before_source, before_target) : Transforms.NaturalCopy<input, axes, plan, cursor>},
897  certificate: Transforms.LoopCertificate<input, axes, plan, original, fuel, 1n+cursor, after_source, after_target>) ->
898  Transforms.LoopCertificate<input, axes, plan, original, 1n+fuel, cursor, before_source, before_target>:
899  Transforms.LoopCertificate{pair, exact, meaning} = certificate
900  Transforms.LoopCertificate{pair,
901    Equal.trans(Core.Tensor<input> & Result<&2, &1, Transforms.Error, Core.Tensor<Layout.output(input, axes, plan)>>, pair,
902      Transforms.copy_loop(fuel, input, axes, plan, after_source, after_target, 1n+cursor),
903      Transforms.copy_loop(1n+fuel, input, axes, plan, before_source, before_target, cursor), exact,
904      Equal.cong(Transforms.NaturalCopy<input, axes, plan, cursor>,
905        Core.Tensor<input> & Result<&2, &1, Transforms.Error, Core.Tensor<Layout.output(input, axes, plan)>>,
906        result => Transforms.natural_next(input, axes, plan, cursor,
907          source => target => Transforms.copy_loop(fuel, input, axes, plan, source, target, 1n+cursor), result),
908        Transforms.NaturalCopied{(after_source, Done{after_target}), index, converted, candidate},
909        Transforms.natural_step(input, axes, plan, cursor, before_source, before_target), natural_exact)),
910    %add_successor_right(fuel, cursor) : Transforms.LoopMeaning(input, axes, plan, original, _, pair)
911    meaning}