diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-20 11:51:14 +0000 |
|---|---|---|
| committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-20 14:49:03 +0000 |
| commit | 7ac07f355f4cb75a54ec423670b7078bd0ecb44d (patch) | |
| tree | 5f28c73decbfe0221c2ecedc204f48a7c00884f0 /src/armnn/QuantizerVisitor.hpp | |
| parent | 2a434a8a23d75fb62ac0cb3ecb83ba7aab89b8c6 (diff) | |
| download | armnn-7ac07f355f4cb75a54ec423670b7078bd0ecb44d.tar.gz | |
IVGCVSW-2858 Add support for QSymm16 quantization
Change-Id: Ia7c305c30c39ec0e9db447a461479be17fde250c
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnn/QuantizerVisitor.hpp')
| -rw-r--r-- | src/armnn/QuantizerVisitor.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp index 2682663047..1751229435 100644 --- a/src/armnn/QuantizerVisitor.hpp +++ b/src/armnn/QuantizerVisitor.hpp @@ -7,6 +7,7 @@ #include "armnn/LayerVisitorBase.hpp" #include "StaticRangeVisitor.hpp" +#include "NetworkQuantizationScheme.hpp" #include <armnn/INetwork.hpp> #include <armnn/Types.hpp> @@ -24,7 +25,7 @@ class StaticRangeVisitor; class QuantizerVisitor : public LayerVisitorBase<VisitorNoThrowPolicy> { public: - QuantizerVisitor(const RangeTracker& rangeTracker); + QuantizerVisitor(const RangeTracker& rangeTracker, const IQuantizationScheme* quantizationScheme); ~QuantizerVisitor() = default; /// Functions to quantize the individual layers, overridden from ILayerVisitor @@ -129,6 +130,8 @@ private: /// Mapping from guid to layer in quantized network std::unordered_map<LayerGuid, IConnectableLayer*> m_QuantizedGuidToLayerMap; + + const IQuantizationScheme* m_QuantizationScheme; }; } //namespace armnn |