diff options
author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-02-27 09:42:06 +0000 |
---|---|---|
committer | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-02-27 09:42:06 +0000 |
commit | b2845655b7814470c2a52cd7d0bee01031615bfc (patch) | |
tree | 9bf986bc681100734ce3de60b1f5dde4c4f3702f /src/armnn/NetworkQuantizerUtils.hpp | |
parent | 7733c57b9bde599d133d8edab49cd7a9069db878 (diff) | |
download | armnn-b2845655b7814470c2a52cd7d0bee01031615bfc.tar.gz |
IVGCVSW-2632 Fix RefMerger from QAsymm8 types with different quantization parameters
Change-Id: Ie67ce4966c5e5fef618876b027292da429de1485
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
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); |