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 44 of file OMKernelExecutionBuilder.h.

Constructor & Destructor Documentation

◆ KernelBuiltinExecuteRegistry()

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

Definition at line 47 of file OMKernelExecutionBuilder.h.

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

Member Function Documentation

◆ getKernelExecuteFunc()

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

Definition at line 59 of file OMKernelExecutionBuilder.h.

60 {
61 const auto builder_id_opcode = size_t(builderID);
62 assert(builder_id_opcode < size_t(core::OMBuilderID::BuiltinOperatorsSize));
63 if (builder_id_opcode >= size_t(core::OMBuilderID::BuiltinOperatorsSize))
64 {
65 *execute_func = nullptr;
66 OM_LOG_AND_RETURN(UnknownError, "Unknown error encountered");
67 }
68 *execute_func = _operator_execute[builder_id_opcode];
69 return Ok;
70 }
#define OM_LOG_AND_RETURN(err, msg)
Definition OMLog.h:31

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

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


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