1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #pragma once 6 7 #include <string> 8 #include <iostream> 9 #include <vector> 10 #include "QuantizationDataSet.hpp" 11 12 namespace armnnQuantizer 13 { 14 15 // parses the command line to extract 16 // * the input file -f containing the serialized fp32 ArmNN input graph (must exist...and be a input graph file) 17 // * the csv file -c <optional> detailing the paths for RAW input tensors to use for refinement 18 // * the directory -d to place the output file into (must already exist and be writable) 19 // * the name of the file -o the quantized ArmNN input graph will be written to (must not already exist) 20 // * LATER: the min and max overrides to be applied to the inputs 21 // specified as -i <int> (input id) -n <float> (minimum) -x <float> (maximum) 22 // multiple sets of -i, -n, -x can appear on the command line but they must match 23 // in number i.e. a -n and -x for each -i and the id of the input must correspond 24 // to an input layer in the fp32 graph when it is loaded. 25 class CommandLineProcessor 26 { 27 public: 28 bool ProcessCommandLine(int argc, char* argv[]); 29 GetInputFileName()30 std::string GetInputFileName() {return m_InputFileName;} GetCsvFileName()31 std::string GetCsvFileName() {return m_CsvFileName;} GetCsvFileDirectory()32 std::string GetCsvFileDirectory() {return m_CsvFileDirectory;} GetOutputDirectoryName()33 std::string GetOutputDirectoryName() {return m_OutputDirectory;} GetOutputFileName()34 std::string GetOutputFileName() {return m_OutputFileName;} GetQuantizationScheme()35 std::string GetQuantizationScheme() {return m_QuantizationScheme;} GetQuantizationDataSet()36 QuantizationDataSet GetQuantizationDataSet() {return m_QuantizationDataSet;} HasPreservedDataType()37 bool HasPreservedDataType() {return m_PreserveDataType;} HasQuantizationData()38 bool HasQuantizationData() {return !m_QuantizationDataSet.IsEmpty();} 39 40 protected: 41 std::string m_InputFileName; 42 std::string m_CsvFileName; 43 std::string m_CsvFileDirectory; 44 std::string m_OutputDirectory; 45 std::string m_OutputFileName; 46 std::string m_QuantizationScheme; 47 QuantizationDataSet m_QuantizationDataSet; 48 bool m_PreserveDataType; 49 }; 50 51 } // namespace armnnQuantizer 52 53