#include <InferenceTest.hpp>

Inheritance diagram for InferenceModelTestCase< TModel >:

Public Member Functions
	InferenceModelTestCase (TModel &model, unsigned int testCaseId, const std::vector< armnnUtils::TContainer > &inputs, const std::vector< unsigned int > &outputSizes)

virtual void	Run () override

Public Member Functions inherited from IInferenceTestCase
virtual	~IInferenceTestCase ()

virtual TestCaseResult	ProcessResult (const InferenceTestOptions &options)=0

Protected Member Functions
unsigned int	GetTestCaseId () const

const std::vector< armnnUtils::TContainer > &	GetOutputs () const

Detailed Description

template<typename TModel>
class armnn::test::InferenceModelTestCase< TModel >

Definition at line 112 of file InferenceTest.hpp.

Constructor & Destructor Documentation

◆ InferenceModelTestCase()

InferenceModelTestCase	(	TModel &	model,
		unsigned int	testCaseId,
		const std::vector< armnnUtils::TContainer > &	inputs,
		const std::vector< unsigned int > &	outputSizes
	)

inline

Definition at line 116 of file InferenceTest.hpp.

         : m_Model(model)
         , m_TestCaseId(testCaseId)
         , m_Inputs(std::move(inputs))
     {
         // Initialize output vector
         const size_t numOutputs = outputSizes.size();
         m_Outputs.reserve(numOutputs);
 
         for (size_t i = 0; i < numOutputs; i++)
         {
             m_Outputs.push_back(std::vector<typename TModel::DataType>(outputSizes[i]));
         }
     }

Member Function Documentation

◆ GetOutputs()

const std::vector<armnnUtils::TContainer>& GetOutputs ( ) const

inlineprotected

Definition at line 141 of file InferenceTest.hpp.

Referenced by ClassifierTestCase< TTestCaseDatabase, TModel >::ProcessResult().

141 { return m_Outputs; }

◆ GetTestCaseId()

unsigned int GetTestCaseId ( ) const

inlineprotected

Definition at line 140 of file InferenceTest.hpp.

Referenced by ClassifierTestCase< TTestCaseDatabase, TModel >::ProcessResult().

140 { return m_TestCaseId; }

◆ Run()

virtual void Run ( )

inlineoverridevirtual

Implements IInferenceTestCase.

Definition at line 134 of file InferenceTest.hpp.

     {
         m_Model.Run(m_Inputs, m_Outputs);
     }

The documentation for this class was generated from the following file:

tests/InferenceTest.hpp

Public Member Functions

Protected Member Functions