armnnQuantizer Namespace Reference

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)

Function Documentation

AddInputData()

void armnnQuantizer::AddInputData	(	unsigned int	passId,
		armnn::LayerBindingId	bindingId,
		const std::string &	inputFilePath,
		std::map< unsigned int, QuantizationInput > &	passIdToQuantizationInput
	)

Definition at line 29 of file QuantizationDataSet.cpp.

Referenced by GetFileNameFromCsvRow().

{
    auto iterator = passIdToQuantizationInput.find(passId);
    if (iterator == passIdToQuantizationInput.end())
    {
        QuantizationInput input(passId, bindingId, inputFilePath);
        passIdToQuantizationInput.emplace(passId, input);
    }
    else
    {
        auto existingQuantizationInput = iterator->second;
        existingQuantizationInput.AddEntry(bindingId, inputFilePath);
    }
}

GetBindingIdFromCsvRow()

armnn::LayerBindingId armnnQuantizer::GetBindingIdFromCsvRow	(	std::vector< armnnUtils::CsvRow >	csvRows,
		unsigned int	rowIndex
	)

Definition at line 88 of file QuantizationDataSet.cpp.

Referenced by GetFileNameFromCsvRow().

{
    armnn::LayerBindingId bindingId;
    try
    {
        bindingId = std::stoi(csvRows[rowIndex].values[1]);
    }
    catch (const std::invalid_argument&)
    {
        throw armnn::ParseException("Binding ID [" + csvRows[rowIndex].values[0] + "]" +
                                    " is not correct format on CSV row " + std::to_string(rowIndex));
    }
    return bindingId;
}

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().

{
    std::string fileName = csvRows[rowIndex].values[2];

    if (!boost::filesystem::exists(fileName))
    {
        throw armnn::ParseException("File [ " + fileName + "] provided on CSV row " + std::to_string(rowIndex) +
                                    " does not exist.");
    }

    if (fileName.empty())
    {
        throw armnn::ParseException("Filename cannot be empty on CSV row " + std::to_string(rowIndex));
    }
    return fileName;
}

GetPassIdFromCsvRow()

unsigned int armnnQuantizer::GetPassIdFromCsvRow	(	std::vector< armnnUtils::CsvRow >	csvRows,
		unsigned int	rowIndex
	)

Definition at line 73 of file QuantizationDataSet.cpp.

Referenced by GetFileNameFromCsvRow().

{
    unsigned int passId;
    try
    {
        passId = static_cast<unsigned int>(std::stoi(csvRows[rowIndex].values[0]));
    }
    catch (const std::invalid_argument&)
    {
        throw armnn::ParseException("Pass ID [" + csvRows[rowIndex].values[0] + "]" +
                                    " is not correct format on CSV row " + std::to_string(rowIndex));
    }
    return passId;
}

ValidateOutputDirectory()

bool armnnQuantizer::ValidateOutputDirectory

(

std::string &

dir

)

Definition at line 17 of file CommandLineProcessor.cpp.

Referenced by CommandLineProcessor::ProcessCommandLine().

{
    if (dir.empty())
    {
        std::cerr << "No output directory specified" << std::endl;
        return false;
    }

    if (dir[dir.length() - 1] != '/')
    {
        dir += "/";
    }

    if (!boost::filesystem::exists(dir))
    {
        std::cerr << "Output directory [" << dir << "] does not exist" << std::endl;
        return false;
    }

    if (!boost::filesystem::is_directory(dir))
    {
        std::cerr << "Given output directory [" << dir << "] is not a directory" << std::endl;
        return false;
    }

    return true;
}

ValidateProvidedFile()

bool armnnQuantizer::ValidateProvidedFile

(

const std::string &

inputFileName

)

Definition at line 45 of file CommandLineProcessor.cpp.

Referenced by CommandLineProcessor::ProcessCommandLine().

{
    if (!boost::filesystem::exists(inputFileName))
    {
        std::cerr << "Provided file [" << inputFileName << "] does not exist" << std::endl;
        return false;
    }

    if (boost::filesystem::is_directory(inputFileName))
    {
        std::cerr << "Given file [" << inputFileName << "] is a directory" << std::endl;
        return false;
    }

    return true;
}

ValidateQuantizationScheme()

bool armnnQuantizer::ValidateQuantizationScheme

(

const std::string &

scheme

)

Definition at line 62 of file CommandLineProcessor.cpp.

Referenced by CommandLineProcessor::ProcessCommandLine().

{
    if (scheme.empty())
    {
        std::cerr << "No Quantization Scheme specified" << std::endl;
        return false;
    }

    std::vector<std::string> supportedSchemes = {
        "QAsymm8",
        "QSymm16"
    };

    auto iterator = std::find(supportedSchemes.begin(), supportedSchemes.end(), scheme);
    if (iterator == supportedSchemes.end())
    {
        std::cerr << "Quantization Scheme [" << scheme << "] is not supported" << std::endl;
        return false;
    }

    return true;
}