ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci_interpreter::test_kernel::TestDataExpandDimsKernel< T > Class Template Reference

#include <ExpandDimsKernel.h>

Collaboration diagram for luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >:

Public Member Functions

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

Protected Attributes

std::vector< T > _input_data
 
std::vector< T > _reference_output_data
 
const unsigned char * _test_kernel_model_circle
 

Detailed Description

template<typename T>
class luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >

Definition at line 123 of file ExpandDimsKernel.h.

Constructor & Destructor Documentation

◆ TestDataExpandDimsKernel()

template<typename T >
luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >::TestDataExpandDimsKernel ( )
inline

Definition at line 126 of file ExpandDimsKernel.h.

References luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >::_input_data, luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >::_reference_output_data, luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >::_test_kernel_model_circle, luci_interpreter::test_kernel::expand_dims_kernel::input_data, luci_interpreter::test_kernel::expand_dims_kernel::reference_output_data, and luci_interpreter::test_kernel::expand_dims_kernel::test_kernel_model_circle.

◆ ~TestDataExpandDimsKernel()

Member Function Documentation

◆ get_input_data_by_index()

template<typename T >
const std::vector< T > & luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >::get_input_data_by_index ( int  i)
inlinefinaloverridevirtual

Implements luci_interpreter::test_kernel::TestDataBase< T, U >.

Definition at line 137 of file ExpandDimsKernel.h.

138 {
139 switch (i)
140 {
141 case 0:
142 return _input_data;
143 default:
144 assert(false && "Wrong input index");
145 }
146 }

References luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >::_input_data.

◆ get_model_ptr()

template<typename T >
const unsigned char * luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >::get_model_ptr ( )
inlinefinaloverridevirtual

◆ get_output_data_by_index()

template<typename T >
const std::vector< T > & luci_interpreter::test_kernel::TestDataExpandDimsKernel< T >::get_output_data_by_index ( int  i)
inlinefinaloverridevirtual

Field Documentation

◆ _input_data

◆ _reference_output_data

◆ _test_kernel_model_circle


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