IVGCVSW-2834 Add dynamic quantization via datasets

* Add QuantizationDataSet class for quantization data parsed from CSV file
* Add QuantizationInput for retrieving quantization data for each layer ID
* Add unit tests for command line processor and QuantizationDataSet

Change-Id: Iaf0a747b5f25a59a766ac04f7158e8cb7909d179
Signed-off-by: Nina Drozd <nina.drozd@arm.com>

1 files changed, 7 insertions, 1 deletions

diff --git a/src/armnnQuantizer/CommandLineProcessor.hpp b/src/armnnQuantizer/CommandLineProcessor.hpp
index 7e366a7664..ae39abb603 100644
--- a/src/armnnQuantizer/CommandLineProcessor.hpp
+++ b/src/armnnQuantizer/CommandLineProcessor.hpp
@@ -6,6 +6,8 @@
 
 #include <string>
 #include <iostream>
+#include <vector>
+#include "QuantizationDataSet.hpp"
 
 namespace armnnQuantizer
 {
@@ -31,13 +33,17 @@ public:
     std::string GetOutputDirectoryName() {return m_OutputDirectory;}
     std::string GetOutputFileName() {return m_OutputFileName;}
     std::string GetQuantizationScheme() {return m_QuantizationScheme;}
-private:
+    QuantizationDataSet GetQuantizationDataSet() {return m_QuantizationDataSet;}
+    bool HasQuantizationData() {return !m_QuantizationDataSet.IsEmpty();}
+
+protected:
     std::string m_InputFileName;
     std::string m_CsvFileName;
     std::string m_CsvFileDirectory;
     std::string m_OutputDirectory;
     std::string m_OutputFileName;
     std::string m_QuantizationScheme;
+    QuantizationDataSet m_QuantizationDataSet;
 };
 
 } // namespace armnnQuantizer