def off_mask

bendtorch-core@0.1.0.0 · src/autograd/reduce_fiber_proof.bend:467

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

467def off_mask(+rest: List<&2, Shape.Row>, +left: Nat, +stride: Nat, +digit: Nat, +target: Nat, +mode: OffMode<rest, left, stride, digit, target>,
468  +q: Nat, +below: {Nat.is_lt(q, stride) == True{} : Bool}) ->
469  {False{} == Nat.is_eq(Nat.add(Nat.mul(left, digit), Shape.walk_left(rest, q)), target) : Bool}:
470  Off{+bound, +differ} = mode
471  Equal.sym(Bool, Nat.is_eq(Nat.add(Nat.mul(left, digit), Shape.walk_left(rest, q)), target), False{},
472    mask_off(left, digit, Shape.walk_left(rest, q), target, bound_at(rest, left, stride, bound, q, below), differ))