diff options
Diffstat (limited to 'src/armnn/NetworkQuantizerUtils.hpp')
-rw-r--r-- | src/armnn/NetworkQuantizerUtils.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/NetworkQuantizerUtils.hpp b/src/armnn/NetworkQuantizerUtils.hpp index 458d21a974..c23517e385 100644 --- a/src/armnn/NetworkQuantizerUtils.hpp +++ b/src/armnn/NetworkQuantizerUtils.hpp @@ -17,7 +17,7 @@ namespace armnn { -std::pair<int, float> ComputeQAsymmParams(int numBits, double min, double max); +std::pair<float, int> ComputeQAsymmParams(int numBits, double min, double max); template<typename srcType> void Quantize(const srcType* src, uint8_t* dst, size_t numElements, float& scale, int& offset) @@ -34,8 +34,8 @@ void Quantize(const srcType* src, uint8_t* dst, size_t numElements, float& scale } auto qParams = ComputeQAsymmParams(8, min, max); - offset = qParams.first; - scale = qParams.second; + scale = qParams.first; + offset = qParams.second; for (size_t i = 0; i < numElements; ++i) { dst[i] = armnn::Quantize<uint8_t>(src[i], scale, offset); |