ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onert_micro::test_model::TestDataArgMaxBase< T, U > Class Template Reference

#include <TestDataArgMaxBase.h>

Collaboration diagram for onert_micro::test_model::TestDataArgMaxBase< T, U >:

Public Member Functions

 TestDataArgMaxBase ()
 
const unsigned char * get_model_ptr () override final
 
const std::vector< T > & get_input_data_by_index (int i) override final
 
const std::vector< U > & get_output_data_by_index (int i) override final
 
- Public Member Functions inherited from onert_micro::test_model::TestDataBase< T, U >
virtual ~TestDataBase ()=default
 

Protected Attributes

std::vector< T > _input1_data
 
std::vector< T > _input2_data
 
std::vector< U > _reference_output_data
 
const unsigned char * _test_arg_max_kernel_model_circle
 

Detailed Description

template<typename T, typename U>
class onert_micro::test_model::TestDataArgMaxBase< T, U >

Definition at line 28 of file TestDataArgMaxBase.h.

Constructor & Destructor Documentation

◆ TestDataArgMaxBase()

template<typename T , typename U >
onert_micro::test_model::TestDataArgMaxBase< T, U >::TestDataArgMaxBase ( )
inlineexplicit

Definition at line 31 of file TestDataArgMaxBase.h.

32 {
33 // Do nothing
34 }

Member Function Documentation

◆ get_input_data_by_index()

template<typename T , typename U >
const std::vector< T > & onert_micro::test_model::TestDataArgMaxBase< T, U >::get_input_data_by_index ( int  i)
inlinefinaloverridevirtual

Implements onert_micro::test_model::TestDataBase< T, U >.

Definition at line 38 of file TestDataArgMaxBase.h.

39 {
40 switch (i)
41 {
42 case 0:
43 return _input1_data;
44 case 1:
45 return _input2_data;
46 default:
47 assert(false && "Wrong input index");
48 }
49 }

References onert_micro::test_model::TestDataArgMaxBase< T, U >::_input1_data, and onert_micro::test_model::TestDataArgMaxBase< T, U >::_input2_data.

◆ get_model_ptr()

template<typename T , typename U >
const unsigned char * onert_micro::test_model::TestDataArgMaxBase< T, U >::get_model_ptr ( )
inlinefinaloverridevirtual

◆ get_output_data_by_index()

template<typename T , typename U >
const std::vector< U > & onert_micro::test_model::TestDataArgMaxBase< T, U >::get_output_data_by_index ( int  i)
inlinefinaloverridevirtual

Field Documentation

◆ _input1_data

template<typename T , typename U >
std::vector<T> onert_micro::test_model::TestDataArgMaxBase< T, U >::_input1_data
protected

◆ _input2_data

template<typename T , typename U >
std::vector<T> onert_micro::test_model::TestDataArgMaxBase< T, U >::_input2_data
protected

◆ _reference_output_data

template<typename T , typename U >
std::vector<U> onert_micro::test_model::TestDataArgMaxBase< T, U >::_reference_output_data
protected

◆ _test_arg_max_kernel_model_circle

template<typename T , typename U >
const unsigned char* onert_micro::test_model::TestDataArgMaxBase< T, U >::_test_arg_max_kernel_model_circle
protected

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