#include <InferenceTest.hpp>

Inheritance diagram for InferenceModelTestCase< TModel >:

Public Types
using	TContainer = mapbox::util::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char > >

Public Member Functions
	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 ()

virtual TestCaseResult	ProcessResult (const InferenceTestOptions &options)=0

Protected Member Functions
unsigned int	GetTestCaseId () const

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

Detailed Description

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

Definition at line 110 of file InferenceTest.hpp.

Member Typedef Documentation

◆ TContainer

using TContainer = mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char> >

Definition at line 113 of file InferenceTest.hpp.

Constructor & Destructor Documentation

◆ InferenceModelTestCase()

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

inline

Definition at line 115 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<TContainer>& GetOutputs ( ) const

inlineprotected

Definition at line 140 of file InferenceTest.hpp.

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

140 { return m_Outputs; }

◆ GetTestCaseId()

unsigned int GetTestCaseId ( ) const

inlineprotected

Definition at line 139 of file InferenceTest.hpp.

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

139 { return m_TestCaseId; }

◆ Run()

virtual void Run ( )

inlineoverridevirtual

Implements IInferenceTestCase.

Definition at line 133 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 Types

Public Member Functions

Protected Member Functions