clamp_max Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/zarch/vec256_zarch.h lines 1173–1179

  template <
      typename U = T,
      std::enable_if_t<std::is_floating_point_v<U>, int> = 0>
  Vectorized<T> clamp_max(const Vectorized<T>& max) const {
    Vectorized<T> tmp = {vec_min(_vec0, max._vec0), vec_min(_vec1, max._vec1)};
    return blendv(tmp, *this, isnan());
  }