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< std::string > tokens, unsigned int lineIndex)
armnn::LayerBindingId	GetBindingIdFromCsvRow (std::vector< std::string > tokens, unsigned int lineIndex)
std::string	GetFileNameFromCsvRow (std::vector< std::string > tokens, unsigned int lineIndex)

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 28 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< std::string >	tokens,
		unsigned int	lineIndex
	)

Definition at line 87 of file QuantizationDataSet.cpp.

Referenced by GetFileNameFromCsvRow().

{
    armnn::LayerBindingId bindingId;
    try
    {
        bindingId = std::stoi(tokens[1]);
    }
    catch (const std::invalid_argument&)
    {
        throw armnn::ParseException(fmt::format("Binding ID [{}] is not correct format on CSV row {}",
                                                tokens[1], lineIndex));
    }
    return bindingId;
}

GetFileNameFromCsvRow()

std::string armnnQuantizer::GetFileNameFromCsvRow	(	std::vector< std::string >	tokens,
		unsigned int	lineIndex
	)

Definition at line 102 of file QuantizationDataSet.cpp.

References AddInputData(), GetBindingIdFromCsvRow(), GetPassIdFromCsvRow(), armnn::stringUtils::StringTokenizer(), and armnn::stringUtils::StringTrim().

{
    std::string fileName = armnn::stringUtils::StringTrim(tokens[2]);

    if (!fs::exists(fileName))
    {
        throw armnn::ParseException(fmt::format("File [{}] provided on CSV row {} does not exist.",
                                                fileName, lineIndex));
    }

    if (fileName.empty())
    {
        throw armnn::ParseException(fmt::format("Filename cannot be empty on CSV row {} ", lineIndex));
    }
    return fileName;
}

GetPassIdFromCsvRow()

unsigned int armnnQuantizer::GetPassIdFromCsvRow	(	std::vector< std::string >	tokens,
		unsigned int	lineIndex
	)

Definition at line 72 of file QuantizationDataSet.cpp.

Referenced by GetFileNameFromCsvRow().

{
    unsigned int passId;
    try
    {
        passId = static_cast<unsigned int>(std::stoi(tokens[0]));
    }
    catch (const std::invalid_argument&)
    {
        throw armnn::ParseException(fmt::format("Pass ID [{}] is not correct format on CSV row {}",
                                                tokens[0], lineIndex));
    }
    return passId;
}

ValidateOutputDirectory()

bool armnnQuantizer::ValidateOutputDirectory

(

std::string &

dir

)

Definition at line 14 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 (!fs::exists(dir))
    {
        std::cerr << "Output directory [" << dir << "] does not exist" << std::endl;
        return false;
    }

    if (!fs::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 42 of file CommandLineProcessor.cpp.

Referenced by CommandLineProcessor::ProcessCommandLine().

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

    if (fs::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 59 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;
}