ArmNN
NotReleased
|
Classes | |
class | CommandLineProcessor |
class | InputLayerVisitor |
Visitor class implementation to gather the TensorInfo for LayerBindingID for creation of ConstTensor for Refine. More... | |
class | QuantizationDataSet |
class | QuantizationInput |
QuantizationInput for specific pass ID, can list a corresponding raw data file for each LayerBindingId. More... | |
Functions | |
bool | ValidateOutputDirectory (std::string &dir) |
bool | ValidateProvidedFile (const std::string &inputFileName) |
bool | ValidateQuantizationScheme (const std::string &scheme) |
void | AddInputData (unsigned int passId, armnn::LayerBindingId bindingId, const std::string &inputFilePath, std::map< unsigned int, QuantizationInput > &passIdToQuantizationInput) |
unsigned int | GetPassIdFromCsvRow (std::vector< armnnUtils::CsvRow > csvRows, unsigned int rowIndex) |
armnn::LayerBindingId | GetBindingIdFromCsvRow (std::vector< armnnUtils::CsvRow > csvRows, unsigned int rowIndex) |
std::string | GetFileNameFromCsvRow (std::vector< armnnUtils::CsvRow > csvRows, unsigned int rowIndex) |
void armnnQuantizer::AddInputData | ( | unsigned int | passId, |
armnn::LayerBindingId | bindingId, | ||
const std::string & | inputFilePath, | ||
std::map< unsigned int, QuantizationInput > & | passIdToQuantizationInput | ||
) |
armnn::LayerBindingId armnnQuantizer::GetBindingIdFromCsvRow | ( | std::vector< armnnUtils::CsvRow > | csvRows, |
unsigned int | rowIndex | ||
) |
Definition at line 88 of file QuantizationDataSet.cpp.
Referenced by GetFileNameFromCsvRow().
std::string armnnQuantizer::GetFileNameFromCsvRow | ( | std::vector< armnnUtils::CsvRow > | csvRows, |
unsigned int | rowIndex | ||
) |
Definition at line 103 of file QuantizationDataSet.cpp.
References AddInputData(), GetBindingIdFromCsvRow(), GetPassIdFromCsvRow(), and CsvReader::ParseFile().
unsigned int armnnQuantizer::GetPassIdFromCsvRow | ( | std::vector< armnnUtils::CsvRow > | csvRows, |
unsigned int | rowIndex | ||
) |
Definition at line 73 of file QuantizationDataSet.cpp.
Referenced by GetFileNameFromCsvRow().
bool armnnQuantizer::ValidateOutputDirectory | ( | std::string & | dir | ) |
Definition at line 17 of file CommandLineProcessor.cpp.
Referenced by CommandLineProcessor::ProcessCommandLine().
bool armnnQuantizer::ValidateProvidedFile | ( | const std::string & | inputFileName | ) |
Definition at line 45 of file CommandLineProcessor.cpp.
Referenced by CommandLineProcessor::ProcessCommandLine().
bool armnnQuantizer::ValidateQuantizationScheme | ( | const std::string & | scheme | ) |
Definition at line 62 of file CommandLineProcessor.cpp.
Referenced by CommandLineProcessor::ProcessCommandLine().