diff options
author | Les Bell <les.bell@arm.com> | 2019-05-17 16:17:12 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-22 10:42:15 +0000 |
commit | e0ca861a5846cb5039b2d4760ce70d03e5efd8f3 (patch) | |
tree | 687b9300a1e9e197c06473d2eac63ab1d3363ffe /src/armnn/NetworkQuantizationScheme.hpp | |
parent | ed735043d679e5697f99badbec9c89fb29e9b6aa (diff) | |
download | armnn-e0ca861a5846cb5039b2d4760ce70d03e5efd8f3.tar.gz |
IVGCVSW-3081 Quantizer min>=max error & missing layers
* relaxed the check to min > max in the quantization schemes
* added missing layer support for resnet_v2_50 model
* Pad, Rsqrt, Multipilcation, Subtraction, Mean
* sorted methods alphabetically in Quantizerlayer & LayerVisitorBase
Change-Id: I003401ff7ac89b60580c959ea8fd9d6fef66b88e
Signed-off-by: Les Bell <les.bell@arm.com>
Diffstat (limited to 'src/armnn/NetworkQuantizationScheme.hpp')
-rw-r--r-- | src/armnn/NetworkQuantizationScheme.hpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/armnn/NetworkQuantizationScheme.hpp b/src/armnn/NetworkQuantizationScheme.hpp index 065205dada..a5f96a106b 100644 --- a/src/armnn/NetworkQuantizationScheme.hpp +++ b/src/armnn/NetworkQuantizationScheme.hpp @@ -30,9 +30,9 @@ struct QAsymm8QuantizationScheme : IQuantizationScheme { OffsetScalePair ComputeScheme(double min, double max) const override { - if (min >= max) + if (min > max) { - throw InvalidArgumentException("min >= max will result in invalid quantization."); + throw InvalidArgumentException("min > max will result in invalid quantization."); } double highest = (1 << NumBits()) - 1; @@ -59,9 +59,9 @@ struct QSymm16QuantizationScheme : IQuantizationScheme { OffsetScalePair ComputeScheme(double min, double max) const override { - if (min >= max) + if (min > max) { - throw InvalidArgumentException("min >= max will result in invalid quantization."); + throw InvalidArgumentException("min > max will result in invalid quantization."); } double highest = (1 << (NumBits()-1)) - 1; // (numbits-1) accounts for the sign bit |