def step_route

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

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

779def step_route(-input: Shape.Shape, -axes: List<&2, U32>, -plan: Layout.Plan<input, axes>, -index: U32,
780  -before_source: Core.Tensor<input>, -before_target: Core.Tensor<Layout.output(input, axes, plan)>,
781  -after_source: Core.Tensor<input>, -after_target: Core.Tensor<Layout.output(input, axes, plan)>,
782  meaning: Transforms.StepMeaning(input, axes, plan, index, before_source, before_target, (after_source, Done{after_target}))) ->
783  Core.ObservationChecked(Core.Packed{Layout.output(input, axes, plan), after_target}, index):
784  (kept, rest) = meaning
785  (checked, values) = rest
786  checked