def mapping_done

bendtorch-core@0.1.0.0 · src/tensor/permute_build_proof.bend:223

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/permutation_mapping.bend as Mapping

223def mapping_done(+input: Shape.Shape, +axes: List<&2, U32>, +plan: Layout.Plan<input, axes>, +index: U32,
224  +inside: {Nat.is_lt(U32.to_nat(index), Shape.logical_numel(Shape.dimensions(Layout.output(input, axes, plan)))) == True{} : Bool}) ->
225  {Result.is_done(&2, &2, Layout.Error, Layout.Mapping<input, axes, plan, index>, Layout.map_index(input, axes, plan, index)) == True{} : Bool}:
226  Mapping.mapping_decided_success(input, axes, plan, index,
227    Shape.bool_case(U32.is_lt(index, Shape.count(Layout.output(input, axes, plan)))),
228    index_bounded(Layout.output(input, axes, plan), index, inside))