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
	QuantizationDataSet is a structure which is created after parsing a quantization CSV file. More...
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 30 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 89 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 104 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 74 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 =
    {
        "QAsymmS8",
        "QAsymmU8",
        "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;
}