From 724e48013142562b7f09c9c819f57c314c4ee3d4 Mon Sep 17 00:00:00 2001
From: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Date: Thu, 9 May 2019 10:13:20 +0100
Subject: 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
---
 src/armnn/QuantizerVisitor.hpp | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'src/armnn/QuantizerVisitor.hpp')

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
-- 
cgit v1.2.1