onert::ir::train::LossInfo Struct Reference

#include <LossInfo.h>

Collaboration diagram for onert::ir::train::LossInfo:

Data Structures
union	LossParam

Public Member Functions
	LossInfo ()

Data Fields
LossCode	loss_code
LossReductionType	reduction_type
union onert::ir::train::LossInfo::LossParam	loss_param

Detailed Description

Definition at line 42 of file LossInfo.h.

Constructor & Destructor Documentation

LossInfo()

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

inline

Definition at line 50 of file LossInfo.h.

    : loss_code{LossCode::Undefined}, reduction_type{LossReductionType::Undefined},
      loss_param{-1, 0.0f}
  {
  }

Field Documentation

loss_code

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

Definition at line 44 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

union onert::ir::train::LossInfo::LossParam onert::ir::train::LossInfo::loss_param

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

reduction_type

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

Definition at line 45 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:

• runtime/onert/core/include/ir/train/LossInfo.h