ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Data Fields
onert_micro::execute::OMExecuteArgs Struct Reference

#include <OMExecuteArgs.h>

Collaboration diagram for onert_micro::execute::OMExecuteArgs:
Collaboration graph
[legend]

Data Fields

core::OMRuntimeStorage & runtime_storage
 
core::OMRuntimeContext & runtime_context
 
uint16_t kernel_index
 
core::OMRuntimeModule & runtime_module
 
uint32_t num_train_layers = 0
 
bool is_train_mode = false
 

Detailed Description

Definition at line 30 of file OMExecuteArgs.h.

Field Documentation

◆ is_train_mode

bool onert_micro::execute::OMExecuteArgs::is_train_mode = false

Definition at line 37 of file OMExecuteArgs.h.

◆ kernel_index

uint16_t onert_micro::execute::OMExecuteArgs::kernel_index

Definition at line 34 of file OMExecuteArgs.h.

Referenced by onert_micro::execute::execute_arg_common(), onert_micro::execute::execute_relu_common(), onert_micro::execute::execute_reshape_common(), onert_micro::execute::readKernelDataTISO(), onert_micro::execute::OMKernelExecute::runForward(), onert_micro::execute::SISOHeader(), and onert_micro::execute::TISOHeader().

◆ num_train_layers

uint32_t onert_micro::execute::OMExecuteArgs::num_train_layers = 0

Definition at line 36 of file OMExecuteArgs.h.

◆ runtime_context

core::OMRuntimeContext& onert_micro::execute::OMExecuteArgs::runtime_context

Definition at line 33 of file OMExecuteArgs.h.

Referenced by onert_micro::execute::execute_arg_common(), onert_micro::execute::execute_relu_common(), onert_micro::execute::execute_reshape_common(), onert_micro::execute::readKernelDataTISO(), onert_micro::execute::OMKernelExecute::runForward(), onert_micro::execute::SISOHeader(), and onert_micro::execute::TISOHeader().

◆ runtime_module

core::OMRuntimeModule& onert_micro::execute::OMExecuteArgs::runtime_module

Definition at line 35 of file OMExecuteArgs.h.

◆ runtime_storage

core::OMRuntimeStorage& onert_micro::execute::OMExecuteArgs::runtime_storage

Definition at line 32 of file OMExecuteArgs.h.

Referenced by onert_micro::execute::execute_arg_common(), onert_micro::execute::execute_relu_common(), onert_micro::execute::execute_reshape_common(), onert_micro::execute::readKernelDataTISO(), onert_micro::execute::OMKernelExecute::runForward(), onert_micro::execute::SISOHeader(), and onert_micro::execute::TISOHeader().

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