def loop_natural_success

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

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
import bendtorch-core@0.1.0.0/src/storage/arith.bend as Arith
import bendtorch-core@0.1.0.0/src/tensor/transform_totality.bend as Totality

1904def loop_natural_success(+fuel: Nat, +input: Shape.Shape, +axes: List<&2, U32>, +plan: Layout.Plan<input, axes>, +cursor: Nat,
1905  result: Transforms.NaturalCopy<input, axes, plan, cursor>,
1906  -successful: {Totality.natural_succeeded(input, axes, plan, cursor, result) == True{} : Bool},
1907  rec: @source: Core.Tensor<input> -> @target: Core.Tensor<Layout.output(input, axes, plan)> ->
1908    Totality.CopySuccess<input, axes, plan, Transforms.copy_loop(fuel, input, axes, plan, source, target, 1n+cursor)>) ->
1909  Totality.CopySuccess<input, axes, plan, Transforms.natural_next(input, axes, plan, cursor,
1910    source => target => Transforms.copy_loop(fuel, input, axes, plan, source, target, 1n+cursor), result)>:
1911  match result:
1912    case Transforms.NaturalError{source}:
1913      Totality.CopySuccess{Empty.absurd({False{} == True{} : Bool}, Arith.clash(successful))}
1914    case Transforms.NaturalCopied{pair, index, exact, candidate}:
1915      loop_pair_success(fuel, input, axes, plan, 1n+cursor, pair, successful, rec)