quantize_tensor_arm Class — pytorch Architecture

Source Code

aten/src/ATen/native/quantized/cpu/kernels/QuantizedOpKernels.cpp lines 3540–3550

template <typename T>
void quantize_tensor_arm(
    const float* __restrict__ in,
    T* __restrict__ out,
    const int64_t N,
    const float scale,
    const int32_t zero_point) {
  for (const auto i : c10::irange(N)) {
    out[i] = at::native::quantize_val<T>(scale, zero_point, in[i]);
  }
}