ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onert_micro::execute::KernelBuiltinExecuteRegistry Class Reference

#include <OMKernelExecutionBuilder.h>

Public Member Functions

constexpr KernelBuiltinExecuteRegistry ()
 
OMStatus getKernelExecuteFunc (core::OMBuilderID builderID, KernelExecuteFunc **execute_func) const
 

Detailed Description

Definition at line 43 of file OMKernelExecutionBuilder.h.

Constructor & Destructor Documentation

◆ KernelBuiltinExecuteRegistry()

constexpr onert_micro::execute::KernelBuiltinExecuteRegistry::KernelBuiltinExecuteRegistry ( )
inlineconstexpr

Definition at line 46 of file OMKernelExecutionBuilder.h.

46 : _operator_execute()
47 {
48#define REGISTER_KERNEL(builtin_operator, name) \
49 registerKernelExecute(core::OMBuilderID::BuiltinOperator_##builtin_operator, \
50 execute_kernel_Circle##name);
51
52#include "KernelsToBuild.lst"
53
54#undef REGISTER_KERNEL
55 }

Member Function Documentation

◆ getKernelExecuteFunc()

OMStatus onert_micro::execute::KernelBuiltinExecuteRegistry::getKernelExecuteFunc ( core::OMBuilderID  builderID,
KernelExecuteFunc **  execute_func 
) const
inline

Definition at line 58 of file OMKernelExecutionBuilder.h.

59 {
60 const auto builder_id_opcode = size_t(builderID);
61 assert(builder_id_opcode < size_t(core::OMBuilderID::BuiltinOperatorsSize));
62 if (builder_id_opcode >= size_t(core::OMBuilderID::BuiltinOperatorsSize))
63 {
64 *execute_func = nullptr;
65 return UnknownError;
66 }
67 *execute_func = _operator_execute[builder_id_opcode];
68 return Ok;
69 }

References onert_micro::core::BuiltinOperatorsSize, onert_micro::Ok, and onert_micro::UnknownError.

Referenced by onert_micro::execute::OMKernelExecute::runForward().


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