1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnnQuantizer/CommandLineProcessor.hpp b/src/armnnQuantizer/CommandLineProcessor.hpp
index f55e7a213f..852fcd4070 100644
--- a/src/armnnQuantizer/CommandLineProcessor.hpp
+++ b/src/armnnQuantizer/CommandLineProcessor.hpp
@@ -12,6 +12,7 @@ namespace armnnQuantizer
 
 // parses the command line to extract
 // * the input file -f containing the serialized fp32 ArmNN input graph (must exist...and be a input graph file)
+// * the csv file -c <optional> detailing the paths for RAW input tensors to use for refinement
 // * the directory -d to place the output file into (must already exist and be writable)
 // * the name of the file -o the quantized ArmNN input graph will be written to (must not already exist)
 // * LATER: the min and max overrides to be applied to the inputs
@@ -25,10 +26,14 @@ public:
     bool ProcessCommandLine(int argc, char* argv[]);
 
     std::string GetInputFileName() {return m_InputFileName;}
+    std::string GetCsvFileName() {return m_CsvFileName;}
+    std::string GetCsvFileDirectory() {return m_CsvFileDirectory;}
     std::string GetOutputDirectoryName() {return m_OutputDirectory;}
     std::string GetOutputFileName() {return m_OutputFileName;}
 private:
     std::string m_InputFileName;
+    std::string m_CsvFileName;
+    std::string m_CsvFileDirectory;
     std::string m_OutputDirectory;
     std::string m_OutputFileName;
 };