ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onert_micro::import Namespace Reference

Namespaces

namespace  helpers
 

Data Structures

class  KernelBuiltinConfigureRegistry
 
class  KernelCustomConfigureRegistry
 
struct  OMConfigureArgs
 
struct  OMDynamicShapesHandler
 
struct  OMExecutionPlanCreator
 
struct  OMKernelConfiguration
 

Typedefs

using KernelConfigureFunc = OMStatus(const OMConfigureArgs &)
 

Functions

OMStatus SISOHeader (const OMConfigureArgs &execute_args, const circle::Tensor **input, const circle::Tensor **output)
 
OMStatus TISOHeader (const OMConfigureArgs &execute_args, const circle::Tensor **input1, const circle::Tensor **input2, const circle::Tensor **output)
 

Variables

constexpr KernelBuiltinConfigureRegistry kernel_builtin_configure
 
constexpr KernelCustomConfigureRegistry kernel_custom_configure
 

Typedef Documentation

◆ KernelConfigureFunc

Function Documentation

◆ SISOHeader()

OMStatus onert_micro::import::SISOHeader ( const OMConfigureArgs execute_args,
const circle::Tensor **  input,
const circle::Tensor **  output 
)

Definition at line 23 of file OMUtils.cpp.

26{
27 OMStatus status;
28
29 core::OMRuntimeContext &runtime_context = execute_args.runtime_context;
30 core::OMRuntimeStorage &runtime_storage = execute_args.runtime_storage;
31 uint16_t op_index = execute_args.kernel_index;
32
33 OMRuntimeKernel runtime_kernel;
34 status = runtime_kernel.readKernel(op_index, runtime_context);
35
36 *input = runtime_kernel.inputs[0];
37 *output = runtime_kernel.outputs[0];
38
39 assert(*input != nullptr);
40 assert(*output != nullptr);
41
42 return status;
43}
OMStatus readKernel(uint16_t op_index, core::OMRuntimeContext &runtime_context)
const circle::Tensor * outputs[maxOutputSize]
const circle::Tensor * inputs[maxInputSize]
core::OMRuntimeContext & runtime_context
core::OMRuntimeStorage & runtime_storage

References onert_micro::execute::OMRuntimeKernel::inputs, onert_micro::import::OMConfigureArgs::kernel_index, onert_micro::execute::OMRuntimeKernel::outputs, onert_micro::execute::OMRuntimeKernel::readKernel(), onert_micro::import::OMConfigureArgs::runtime_context, and onert_micro::import::OMConfigureArgs::runtime_storage.

◆ TISOHeader()

OMStatus onert_micro::import::TISOHeader ( const OMConfigureArgs execute_args,
const circle::Tensor **  input1,
const circle::Tensor **  input2,
const circle::Tensor **  output 
)

Definition at line 45 of file OMUtils.cpp.

49{
50 OMStatus status;
51
52 core::OMRuntimeContext &runtime_context = execute_args.runtime_context;
53 core::OMRuntimeStorage &runtime_storage = execute_args.runtime_storage;
54 uint16_t op_index = execute_args.kernel_index;
55
56 OMRuntimeKernel runtime_kernel;
57 status = runtime_kernel.readKernel(op_index, runtime_context);
58
59 *input2 = runtime_kernel.inputs[0];
60 *input1 = runtime_kernel.inputs[1];
61 *output = runtime_kernel.outputs[0];
62
63 assert(*input1 != nullptr);
64 assert(*input2 != nullptr);
65 assert(*output != nullptr);
66
67 return status;
68}

References onert_micro::execute::OMRuntimeKernel::inputs, onert_micro::import::OMConfigureArgs::kernel_index, onert_micro::execute::OMRuntimeKernel::outputs, onert_micro::execute::OMRuntimeKernel::readKernel(), onert_micro::import::OMConfigureArgs::runtime_context, and onert_micro::import::OMConfigureArgs::runtime_storage.

Variable Documentation

◆ kernel_builtin_configure

constexpr KernelBuiltinConfigureRegistry onert_micro::import::kernel_builtin_configure
constexpr

◆ kernel_custom_configure

constexpr KernelCustomConfigureRegistry onert_micro::import::kernel_custom_configure
constexpr