aboutsummaryrefslogtreecommitdiff
path: root/src/armnnQuantizer/CommandLineProcessor.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnQuantizer/CommandLineProcessor.cpp')
-rw-r--r--src/armnnQuantizer/CommandLineProcessor.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnnQuantizer/CommandLineProcessor.cpp b/src/armnnQuantizer/CommandLineProcessor.cpp
index 4f0d989d3f..758726e72a 100644
--- a/src/armnnQuantizer/CommandLineProcessor.cpp
+++ b/src/armnnQuantizer/CommandLineProcessor.cpp
@@ -97,6 +97,8 @@ bool CommandLineProcessor::ProcessCommandLine(int argc, char* argv[])
"Quantization scheme, \"QAsymm8\" or \"QSymm16\", default value QAsymm8")
("csvfile,c", po::value<std::string>(&m_CsvFileName)->default_value(""),
"CSV file containing paths for RAW input tensors")
+ ("preserve-data-type,p", po::bool_switch(&m_PreserveDataType)->default_value(false),
+ "Preserve the input and output data types")
("outdir,d", po::value<std::string>(&m_OutputDirectory)->required(),
"Directory that output file will be written to")
("outfile,o", po::value<std::string>(&m_OutputFileName)->required(), "Output file name");