InferenceTest.hpp File Reference

#include "InferenceModel.hpp"
#include <armnn/ArmNN.hpp>
#include <armnn/Logging.hpp>
#include <armnn/TypesUtils.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <cxxopts/cxxopts.hpp>
#include <fmt/format.h>
#include "InferenceTest.inl"

Go to the source code of this file.

Classes
class	TestFrameworkException
struct	InferenceTestOptions
class	IInferenceTestCase
class	IInferenceTestCaseProvider
class	InferenceModelTestCase< TModel >
class	ClassifierTestCase< TTestCaseDatabase, TModel >
class	ClassifierTestCaseProvider< TDatabase, InferenceModel >

Namespaces
	armnn
	Copyright (c) 2020 ARM Limited.
	armnn::test

Enumerations
enum	TestCaseResult { Ok, Failed, Abort }

Functions
std::istream &	operator>> (std::istream &in, armnn::Compute &compute)
std::istream &	operator>> (std::istream &in, armnn::BackendId &backend)
bool	ParseCommandLine (int argc, char **argv, IInferenceTestCaseProvider &testCaseProvider, InferenceTestOptions &outParams)
	Parse the command line of an ArmNN (or referencetests) inference test program. More...
bool	ValidateDirectory (std::string &dir)
bool	InferenceTest (const InferenceTestOptions &params, const std::vector< unsigned int > &defaultTestCaseIds, IInferenceTestCaseProvider &testCaseProvider)
template<typename TConstructTestCaseProvider >
int	InferenceTestMain (int argc, char *argv[], const std::vector< unsigned int > &defaultTestCaseIds, TConstructTestCaseProvider constructTestCaseProvider)
template<typename TDatabase , typename TParser , typename TConstructDatabaseCallable >
int	ClassifierInferenceTestMain (int argc, char *argv[], const char *modelFilename, bool isModelBinary, const char *inputBindingName, const char *outputBindingName, const std::vector< unsigned int > &defaultTestCaseIds, TConstructDatabaseCallable constructDatabase, const armnn::TensorShape *inputTensorShape=nullptr)