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

#include <KernelBuilder.h>

Public Types

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

Public Member Functions

constexpr KernelConfigureRegistry ()
 
void configure_kernel (const circle::Operator *cur_op, circle::BuiltinOperator opcode, BaseRuntimeGraph *runtime_graph) const
 

Detailed Description

Definition at line 63 of file KernelBuilder.h.

Member Typedef Documentation

◆ KernelConfigureFunc

Definition at line 66 of file KernelBuilder.h.

Constructor & Destructor Documentation

◆ KernelConfigureRegistry()

constexpr luci_interpreter::KernelConfigureRegistry::KernelConfigureRegistry ( )
inlineconstexpr

Definition at line 68 of file KernelBuilder.h.

68 : _operator_configure()
69 {
70#define REGISTER_KERNEL(builtin_operator, name) \
71 register_kernel_configure(BuilderID::BuiltinOperator_##builtin_operator, \
72 configure_kernel_Circle##name);
73
74#if USE_GENERATED_LIST
75#include "GeneratedKernelsToBuild.lst"
76#else
77#include "KernelsToBuild.lst"
78#endif
79
80#undef REGISTER_KERNEL
81 }

Member Function Documentation

◆ configure_kernel()

void luci_interpreter::KernelConfigureRegistry::configure_kernel ( const circle::Operator *  cur_op,
circle::BuiltinOperator  opcode,
BaseRuntimeGraph runtime_graph 
) const

Definition at line 22 of file KernelBuilder.cpp.

25{
26 auto specific_configure_func = get_kernel_configure_func(opcode);
27 if (specific_configure_func == nullptr)
28 assert(false && "Unsupported operator");
29
30 specific_configure_func(cur_op, runtime_graph);
31}

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


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