mask Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec128/vec128_reduced_precision_common_neon.h lines 53–75

  template <int64_t mask>
  static Derived blend(const Derived& a, const Derived& b) {
    Derived vec;
    vec.values = BlendRegs < 0,
    (mask & 0x01) != 0 > ::impl(a.values, b.values, vec.values);
    vec.values = BlendRegs < 1,
    (mask & 0x02) != 0 > ::impl(a.values, b.values, vec.values);
    vec.values = BlendRegs < 2,
    (mask & 0x04) != 0 > ::impl(a.values, b.values, vec.values);
    vec.values = BlendRegs < 3,
    (mask & 0x08) != 0 > ::impl(a.values, b.values, vec.values);

    vec.values = BlendRegs < 4,
    (mask & 0x10) != 0 > ::impl(a.values, b.values, vec.values);
    vec.values = BlendRegs < 5,
    (mask & 0x20) != 0 > ::impl(a.values, b.values, vec.values);
    vec.values = BlendRegs < 6,
    (mask & 0x40) != 0 > ::impl(a.values, b.values, vec.values);
    vec.values = BlendRegs < 7,
    (mask & 0x80) != 0 > ::impl(a.values, b.values, vec.values);

    return vec;
  }