ArmNN
NotReleased
|
#include <InferenceTest.hpp>
Public Member Functions | |
ClassifierTestCase (int &numInferencesRef, int &numCorrectInferencesRef, const std::vector< unsigned int > &validationPredictions, std::vector< unsigned int > *validationPredictionsOut, TModel &model, unsigned int testCaseId, unsigned int label, std::vector< typename TModel::DataType > modelInput) | |
virtual TestCaseResult | ProcessResult (const InferenceTestOptions ¶ms) override |
Public Member Functions inherited from InferenceModelTestCase< TModel > | |
InferenceModelTestCase (TModel &model, unsigned int testCaseId, const std::vector< TContainer > &inputs, const std::vector< unsigned int > &outputSizes) | |
virtual void | Run () override |
Public Member Functions inherited from IInferenceTestCase | |
virtual | ~IInferenceTestCase () |
Additional Inherited Members | |
Public Types inherited from InferenceModelTestCase< TModel > | |
using | TContainer = boost::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char > > |
Protected Member Functions inherited from InferenceModelTestCase< TModel > | |
unsigned int | GetTestCaseId () const |
const std::vector< TContainer > & | GetOutputs () const |
Definition at line 148 of file InferenceTest.hpp.
ClassifierTestCase | ( | int & | numInferencesRef, |
int & | numCorrectInferencesRef, | ||
const std::vector< unsigned int > & | validationPredictions, | ||
std::vector< unsigned int > * | validationPredictionsOut, | ||
TModel & | model, | ||
unsigned int | testCaseId, | ||
unsigned int | label, | ||
std::vector< typename TModel::DataType > | modelInput | ||
) |
|
overridevirtual |
Implements IInferenceTestCase.
Definition at line 116 of file InferenceTest.inl.
References ARMNN_LOG, armnn::error, armnn::test::Failed, InferenceModelTestCase< TModel >::GetOutputs(), InferenceModelTestCase< TModel >::GetTestCaseId(), armnn::info, InferenceTestOptions::m_IterationCount, and armnn::test::Ok.