blob: 9de03fb327847e869d59d4f7d979dc9fcfe6772d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
//
// 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
|
