def mask_off

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

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

222def mask_off(+size: Nat, +d: Nat, +w: Nat, +t: Nat, below: {Nat.is_lt(w, size) == True{} : Bool},
223  differ: {Nat.is_eq(d, Nat.div(t, size)) == False{} : Bool}) ->
224  {Nat.is_eq(Nat.add(Nat.mul(size, d), w), t) == False{} : Bool}:
225  mask_off_decided(size, d, w, t, below, differ, Shape.bool_case(Nat.is_eq(Nat.add(Nat.mul(size, d), w), t)))