diff options
Diffstat (limited to 'src/armnnQuantizer/CommandLineProcessor.hpp')
-rw-r--r-- | src/armnnQuantizer/CommandLineProcessor.hpp | 52 |
1 files changed, 0 insertions, 52 deletions
diff --git a/src/armnnQuantizer/CommandLineProcessor.hpp b/src/armnnQuantizer/CommandLineProcessor.hpp deleted file mode 100644 index 9de03fb327..0000000000 --- a/src/armnnQuantizer/CommandLineProcessor.hpp +++ /dev/null @@ -1,52 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#pragma once - -#include <string> -#include <iostream> -#include <vector> -#include "QuantizationDataSet.hpp" - -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 -// specified as -i <int> (input id) -n <float> (minimum) -x <float> (maximum) -// multiple sets of -i, -n, -x can appear on the command line but they must match -// in number i.e. a -n and -x for each -i and the id of the input must correspond -// to an input layer in the fp32 graph when it is loaded. -class CommandLineProcessor -{ -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;} - std::string GetQuantizationScheme() {return m_QuantizationScheme;} - QuantizationDataSet GetQuantizationDataSet() {return m_QuantizationDataSet;} - bool HasPreservedDataType() {return m_PreserveDataType;} - 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; - bool m_PreserveDataType; -}; - -} // namespace armnnQuantizer - |