law primitive_source_mapping

bendtorch-core@0.1.0.0 · src/tensor/transform_LAWS.bend:140

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/storage/model.bend as Model
import bendtorch-core@0.1.0.0/src/storage/refinement.bend as Refinement

140law primitive_source_mapping:
141  for -input: Shape.Shape
142  for -axes: List<&2, U32>
143  for -plan: Layout.Plan<input, axes>
144  for -output_index: U32
145  for +mapping: Layout.Mapping<input, axes, plan, output_index>
146  for +depth: Nat
147  for array: Array<F32>
148  for balanced: Model.Balanced(F32, depth, array)
149  {Some{Refinement.second(F32, Array.get(F32, array, Layout.source_index(input, axes, plan, output_index, mapping)))} ==
150    Model.observe(F32, depth, array, Refinement.route(depth, Layout.source_index(input, axes, plan, output_index, mapping))) : Maybe<&2, F32>}