convert_float_to_int8 Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/zarch/vec256_zarch.h lines 2953–2967

template <typename T>
std::enable_if_t<
    std::is_same_v<T, uint8_t>,
    at::vec::Vectorized<
        T>> inline convert_float_to_int8(const Vectorized<float>& src) {
  constexpr auto min_val = std::numeric_limits<T>::min();
  constexpr auto max_val = std::numeric_limits<T>::max();

  auto vec_int = clamp(
      zvec_convert_to_int(src),
      Vectorized<int32_t>(min_val),
      Vectorized<int32_t>(max_val));

  return vec_int.to_vec_uint8_helper();
}