20.02
|
Classes | |
class | ClassifierTestCase |
class | ClassifierTestCaseProvider |
class | IInferenceTestCase |
class | IInferenceTestCaseProvider |
class | InferenceModelTestCase |
struct | InferenceTestOptions |
class | TestFrameworkException |
Typedefs | |
using | TContainer = boost::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char > > |
Enumerations | |
enum | TestCaseResult { Ok, Failed, Abort } |
Functions | |
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 ¶ms, 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) |
using TContainer = boost::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char> > |
Definition at line 30 of file InferenceTest.inl.
|
strong |
Definition at line 70 of file InferenceTest.hpp.
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 |
||
) |
Definition at line 352 of file InferenceTest.inl.
References InferenceTestMain(), Params::m_ModelPath, and ValidateDirectory().
Referenced by main().
bool InferenceTest | ( | const InferenceTestOptions & | params, |
const std::vector< unsigned int > & | defaultTestCaseIds, | ||
IInferenceTestCaseProvider & | testCaseProvider | ||
) |
Definition at line 120 of file InferenceTest.cpp.
References ARMNN_LOG, armnn::error, IInferenceTestCaseProvider::GetTestCase(), armnn::info, InferenceTestOptions::m_EnableProfiling, InferenceTestOptions::m_InferenceTimesFile, InferenceTestOptions::m_IterationCount, IInferenceTestCaseProvider::OnInferenceTestFinished(), armnn::warning, and Exception::what().
Referenced by InferenceTestMain().
int InferenceTestMain | ( | int | argc, |
char * | argv[], | ||
const std::vector< unsigned int > & | defaultTestCaseIds, | ||
TConstructTestCaseProvider | constructTestCaseProvider | ||
) |
Definition at line 301 of file InferenceTest.inl.
References ARMNN_LOG, armnn::ConfigureLogging(), armnn::Debug, armnn::fatal, InferenceTest(), armnn::Info, ParseCommandLine(), and Exception::what().
Referenced by ClassifierInferenceTestMain(), and main().
bool ParseCommandLine | ( | int | argc, |
char ** | argv, | ||
IInferenceTestCaseProvider & | testCaseProvider, | ||
InferenceTestOptions & | outParams | ||
) |
Parse the command line of an ArmNN (or referencetests) inference test program.
Definition at line 31 of file InferenceTest.cpp.
References IInferenceTestCaseProvider::AddCommandLineOptions(), InferenceTestOptions::m_EnableProfiling, InferenceTestOptions::m_InferenceTimesFile, InferenceTestOptions::m_IterationCount, and IInferenceTestCaseProvider::ProcessCommandLineOptions().
Referenced by InferenceTestMain().
bool ValidateDirectory | ( | std::string & | dir | ) |
Definition at line 92 of file InferenceTest.cpp.
Referenced by ClassifierInferenceTestMain(), main(), ClassifierTestCaseProvider< TDatabase, InferenceModel >::ProcessCommandLineOptions(), and YoloTestCaseProvider< Model >::ProcessCommandLineOptions().