def output_below

bendtorch-core@0.1.0.0 · src/autograd/broadcast_pointwise_PROOF.bend:139

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

139def output_below(+plan: Shape.BroadcastPlan, +target: Shape.Shape, +index: U32,
140  same: {Shape.dimensions(Shape.output_shape(plan)) == Shape.dimensions(target) : List<&2, U32>},
141  inside: {Nat.is_lt(U32.to_nat(index), Shape.logical_numel(Shape.dimensions(target))) == True{} : Bool}) ->
142  {Nat.is_lt(U32.to_nat(index), Shape.logical_numel(Shape.dimensions(Shape.output_shape(plan)))) == True{} : Bool}:
143  %Equal.sym(List<&2, U32>, Shape.dimensions(Shape.output_shape(plan)), Shape.dimensions(target), same) :
144    {Nat.is_lt(U32.to_nat(index), Shape.logical_numel(_)) == True{} : Bool}
145  inside