ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci_interpreter::KernelExecuteRegistry Class Reference

#include <KernelBuilder.h>

Public Types

using KernelExecuteFunc = void(const circle::Operator *, BaseRuntimeGraph *)
 

Public Member Functions

constexpr KernelExecuteRegistry ()
 
void execute_kernel (const circle::Operator *cur_op, circle::BuiltinOperator opcode, BaseRuntimeGraph *runtime_graph) const
 

Detailed Description

Definition at line 104 of file KernelBuilder.h.

Member Typedef Documentation

◆ KernelExecuteFunc

Definition at line 107 of file KernelBuilder.h.

Constructor & Destructor Documentation

◆ KernelExecuteRegistry()

constexpr luci_interpreter::KernelExecuteRegistry::KernelExecuteRegistry ( )
inlineconstexpr

Definition at line 109 of file KernelBuilder.h.

109 : _operator_execute()
110 {
111#define REGISTER_KERNEL(builtin_operator, name) \
112 register_kernel_execute(BuilderID::BuiltinOperator_##builtin_operator, \
113 execute_kernel_Circle##name);
114
115#if USE_GENERATED_LIST
116#include "GeneratedKernelsToBuild.lst"
117#else
118#include "KernelsToBuild.lst"
119#endif
120
121#undef REGISTER_KERNEL
122 }

Member Function Documentation

◆ execute_kernel()

void luci_interpreter::KernelExecuteRegistry::execute_kernel ( const circle::Operator *  cur_op,
circle::BuiltinOperator  opcode,
BaseRuntimeGraph runtime_graph 
) const

Definition at line 33 of file KernelBuilder.cpp.

36{
37 auto specific_execute_func = get_kernel_execute_func(opcode);
38 if (specific_execute_func == nullptr)
39 assert(false && "Unsupported operator");
40
41 specific_execute_func(cur_op, runtime_graph);
42}

Referenced by luci_interpreter::RuntimeGraph::execute().


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