commit 5696bff29143667ffd69856e4939cecfbcbd03b3
author Éanna Ó Catháin <eanna.ocathain@arm.com> Fri May 10 13:29:13 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Fri May 10 14:29:36 2019 +0000
tree 77e51754e25b83747b7b8d1dd43a050bdd79bccc
parent 5200afa2742ad9cd1cda7fbce8604794c0616818

IVGCVSW-3063 Modify Quantizer tool to take command line input
             for preserving input/output types

Change-Id: Ib30f1e2e58407526cd881fec33f45fc4e4ed4afe
Signed-off-by: Éanna Ó Catháin <eanna.ocathain@arm.com>

src/armnnQuantizer/CommandLineProcessor.hpp

diff --git a/src/armnnQuantizer/CommandLineProcessor.hpp b/src/armnnQuantizer/CommandLineProcessor.hpp
index ae39abb..9de03fb 100644
--- a/src/armnnQuantizer/CommandLineProcessor.hpp
+++ b/src/armnnQuantizer/CommandLineProcessor.hpp
@@ -34,6 +34,7 @@
     std::string GetOutputFileName() {return m_OutputFileName;}
     std::string GetQuantizationScheme() {return m_QuantizationScheme;}
     QuantizationDataSet GetQuantizationDataSet() {return m_QuantizationDataSet;}
+    bool HasPreservedDataType() {return m_PreserveDataType;}
     bool HasQuantizationData() {return !m_QuantizationDataSet.IsEmpty();}
 
 protected:
@@ -44,6 +45,7 @@
     std::string m_OutputFileName;
     std::string m_QuantizationScheme;
     QuantizationDataSet m_QuantizationDataSet;
+    bool m_PreserveDataType;
 };
 
 } // namespace armnnQuantizer