diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-05-09 10:13:20 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-10 08:30:55 +0000 |
commit | 724e48013142562b7f09c9c819f57c314c4ee3d4 (patch) | |
tree | 935b425961ccb13a488708312bcc70b3b32fc87e /src/armnn/QuantizerVisitor.hpp | |
parent | 5fa83938592db420914903235daf3f1d5c97d6bc (diff) | |
download | armnn-724e48013142562b7f09c9c819f57c314c4ee3d4.tar.gz |
IVGCVSW-3061 Modify NetworkQuantizer to support option to preserve input/output types
* Also add unit tests for new preserve type option
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Change-Id: I860759072f2e3546698118d1bcd5e79eb4e805ec
Diffstat (limited to 'src/armnn/QuantizerVisitor.hpp')
-rw-r--r-- | src/armnn/QuantizerVisitor.hpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp index eb9ebac3d9..300ac164de 100644 --- a/src/armnn/QuantizerVisitor.hpp +++ b/src/armnn/QuantizerVisitor.hpp @@ -25,7 +25,10 @@ class StaticRangeVisitor; class QuantizerVisitor : public LayerVisitorBase<VisitorNoThrowPolicy> { public: - QuantizerVisitor(const RangeTracker& rangeTracker, const IQuantizationScheme* quantizationScheme); + QuantizerVisitor(const RangeTracker& rangeTracker, + const IQuantizationScheme* quantizationScheme, + bool preserveType = false); + ~QuantizerVisitor() = default; /// Functions to quantize the individual layers, overridden from ILayerVisitor @@ -132,6 +135,8 @@ private: std::unordered_map<LayerGuid, IConnectableLayer*> m_QuantizedGuidToLayerMap; const IQuantizationScheme* m_QuantizationScheme; + + const bool m_PreserveType; }; } //namespace armnn |