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

#include <OMBackpropExecuteArgs.h>

Collaboration diagram for onert_micro::train::OMBackpropExecuteArgs:
Collaboration graph
[legend]

Data Fields

core::OMRuntimeStorage & forward_storage
 
core::OMRuntimeStorage & backward_storage
 
core::OMRuntimeContext & backward_context
 
bool is_last_layer
 
bool is_trainable_layer
 
uint16_t kernel_index
 
core::OpTrainableRankType train_rank_type
 

Detailed Description

Definition at line 34 of file OMBackpropExecuteArgs.h.

Field Documentation

◆ backward_context

core::OMRuntimeContext& onert_micro::train::OMBackpropExecuteArgs::backward_context

Definition at line 38 of file OMBackpropExecuteArgs.h.

◆ backward_storage

core::OMRuntimeStorage& onert_micro::train::OMBackpropExecuteArgs::backward_storage

Definition at line 37 of file OMBackpropExecuteArgs.h.

◆ forward_storage

core::OMRuntimeStorage& onert_micro::train::OMBackpropExecuteArgs::forward_storage

Definition at line 36 of file OMBackpropExecuteArgs.h.

◆ is_last_layer

bool onert_micro::train::OMBackpropExecuteArgs::is_last_layer

Definition at line 39 of file OMBackpropExecuteArgs.h.

◆ is_trainable_layer

bool onert_micro::train::OMBackpropExecuteArgs::is_trainable_layer

Definition at line 40 of file OMBackpropExecuteArgs.h.

◆ kernel_index

uint16_t onert_micro::train::OMBackpropExecuteArgs::kernel_index

Definition at line 41 of file OMBackpropExecuteArgs.h.

◆ train_rank_type

core::OpTrainableRankType onert_micro::train::OMBackpropExecuteArgs::train_rank_type

Definition at line 42 of file OMBackpropExecuteArgs.h.

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