diff options
Diffstat (limited to 'src/armnnQuantizer/CommandLineProcessor.hpp')
-rw-r--r-- | src/armnnQuantizer/CommandLineProcessor.hpp | 8 |
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 |