diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-04-12 15:17:02 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-04-12 15:17:02 +0100 |
commit | 2b03d64da2a3a39d069a7a2366f14439afb1ad39 (patch) | |
tree | 7a6ed87ad3c9feeea779fa4e241d6f589cb0dc56 /src/armnnQuantizer/CommandLineProcessor.hpp | |
parent | 7b4886faccb52af9afe7fdeffcbae87e7fbc1484 (diff) | |
download | armnn-2b03d64da2a3a39d069a7a2366f14439afb1ad39.tar.gz |
IVGCVSW-2955 Update the Quantizer Tool to take an additional parameter for the user to specify a CSV file
Change-Id: Id56e09f147cca5c1301ec1b6bac656cd50bfd583
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Diffstat (limited to 'src/armnnQuantizer/CommandLineProcessor.hpp')
-rw-r--r-- | src/armnnQuantizer/CommandLineProcessor.hpp | 5 |
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; }; |