convert_int8_to_float Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/zarch/vec256_zarch.h lines 2941–2951

template <typename T>
std::enable_if_t<
    std::is_same_v<T, uint8_t>,
    at::vec::Vectorized<
        float>> inline convert_int8_to_float(const Vectorized<T>& src) {
  // Note: this function only convert inputs number of elements equal to
  // at::vec::Vectorized<float>.size() Only handle first 64 bits
  auto vec_int = src.to_vec_float_helper();

  return zvec_convert_to_float(vec_int);
}