ONE - On-device Neural Engine
•All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Public Member Functions | Data Fields
onert::ir::train::LossInfo Struct Reference

#include <LossInfo.h>

Public Member Functions

 LossInfo ()
 

Data Fields

LossCode loss_code
 
LossReductionType reduction_type
 
std::variant< std::monostate, CategoricalCrossentropyParam > loss_param
 

Detailed Description

Definition at line 41 of file LossInfo.h.

Constructor & Destructor Documentation

◆ LossInfo()

onert::ir::train::LossInfo::LossInfo ( )
inline

Definition at line 47 of file LossInfo.h.

48 : loss_code{LossCode::Undefined}, reduction_type{LossReductionType::Undefined},
49 loss_param{CategoricalCrossentropyParam{-1, 0.0f}}
50 {
51 }
onert::ir::train::LossInfo::reduction_type
LossReductionType reduction_type
Definition LossInfo.h:44
onert::ir::train::LossInfo::loss_param
std::variant< std::monostate, CategoricalCrossentropyParam > loss_param
Definition LossInfo.h:45

Field Documentation

◆ loss_code

LossCode onert::ir::train::LossInfo::loss_code

Definition at line 43 of file LossInfo.h.

Referenced by onert::ir::train::TrainingInfo::isValid(), onert::ir::train::operation::Loss::name(), and onert::backend::train::KernelGenerator::visit().

◆ loss_param

std::variant<std::monostate, CategoricalCrossentropyParam> onert::ir::train::LossInfo::loss_param

Definition at line 45 of file LossInfo.h.

Referenced by onert::backend::train::KernelGenerator::visit().

◆ reduction_type

LossReductionType onert::ir::train::LossInfo::reduction_type

Definition at line 44 of file LossInfo.h.

Referenced by onert::exec::train::TrainableExecutor::getLoss(), onert::ir::train::TrainingInfo::isValid(), and onert::backend::train::KernelGenerator::visit().

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