precision Class — pytorch Architecture

Source Code

aten/src/ATen/native/quantized/AffineQuantizerBase.cpp lines 56–69

template <typename T, int precision>
void quantize_vec(
    double scale,
    int64_t zero_point,
    const float* src,
    T* dst,
    size_t count) {
  fbgemm::Quantize<typename T::underlying, false /*LEGACY*/>(
      src,
      (typename T::underlying*)dst,
      count,
      fbgemm::TensorQuantizationParams{
          static_cast<float>(scale), static_cast<int32_t>(zero_point), precision});
}