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/NetworkQuantizer.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/NetworkQuantizer.hpp')
-rw-r--r-- | src/armnn/NetworkQuantizer.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnn/NetworkQuantizer.hpp b/src/armnn/NetworkQuantizer.hpp index 5b87851edf..5e93f70290 100644 --- a/src/armnn/NetworkQuantizer.hpp +++ b/src/armnn/NetworkQuantizer.hpp @@ -17,7 +17,8 @@ namespace armnn class NetworkQuantizer : public INetworkQuantizer { public: - NetworkQuantizer(INetwork* inputNetwork) : m_InputNetwork(inputNetwork) {} + NetworkQuantizer(INetwork* inputNetwork, const QuantizerOptions& options) + : m_InputNetwork(inputNetwork), m_Options(options) {} void OverrideInputRange(LayerBindingId layerId, float min, float max) override; INetworkPtr ExportNetwork() override; @@ -28,6 +29,9 @@ private: /// Mapping from Guid to an array of ranges for outputs RangeTracker m_Ranges; + + /// Options for the NetworkQuantizer + QuantizerOptions m_Options; }; } //namespace armnn |