onert::ir::train::TrainingInfo Class Reference

#include <TrainingInfo.h>

Public Member Functions
	TrainingInfo ()
	TrainingInfo (const TrainingInfo &)=default
	TrainingInfo (TrainingInfo &&)=default
TrainingInfo &	operator= (const TrainingInfo &)=default
TrainingInfo &	operator= (TrainingInfo &&)=default
	~TrainingInfo ()=default
uint32_t	version () const
const LossInfo &	lossInfo () const
const OptimizerInfo &	optimizerInfo () const
uint32_t	batchSize () const
const uint32_t &	trainingStep () const
const std::set< OperationIndex > &	getTrainableOps () const
void	setVersion (const uint32_t version)
void	setBatchSize (const uint32_t batch_size)
void	setLossInfo (const LossInfo &loss_info)
void	setOptimizerInfo (const OptimizerInfo &optimizer_info)
uint32_t &	trainingStep ()
void	setTrainableOps (const std::set< OperationIndex > &trainable_ops)
bool	isValid () const

Detailed Description

Definition at line 31 of file TrainingInfo.h.

Constructor & Destructor Documentation

TrainingInfo() [1/3]

onert::ir::train::TrainingInfo::TrainingInfo ( )

inline

Definition at line 34 of file TrainingInfo.h.

    : _version{0}, _loss_info(), _optimizer_info(), _batch_size(0), _training_step{0},
      _trainable_ops{}
  {
  }

TrainingInfo() [2/3]

onert::ir::train::TrainingInfo::TrainingInfo ( const TrainingInfo &  )

default

TrainingInfo() [3/3]

onert::ir::train::TrainingInfo::TrainingInfo ( TrainingInfo &&  )

default

~TrainingInfo()

onert::ir::train::TrainingInfo::~TrainingInfo ( )

default

Member Function Documentation

batchSize()

uint32_t onert::ir::train::TrainingInfo::batchSize ( ) const

inline

Definition at line 49 of file TrainingInfo.h.

{ return _batch_size; }

Referenced by onert::compiler::train::TrainingCompiler::compile().

getTrainableOps()

const std::set< OperationIndex > & onert::ir::train::TrainingInfo::getTrainableOps ( ) const

inline

Definition at line 51 of file TrainingInfo.h.

{ return _trainable_ops; }

Referenced by onert::compiler::train::TrainingCompiler::compile().

isValid()

bool onert::ir::train::TrainingInfo::isValid

(

)

const

Definition at line 22 of file TrainingInfo.cc.

{
  if (_batch_size == 0)
    return false;
 
  if (_optimizer_info.optim_code == OptimizerCode::Undefined)
    return false;
 
  if (_optimizer_info.learning_rate <= 0.0f)
    return false;
 
  if (_loss_info.loss_code == LossCode::Undefined)
    return false;
 
  if (_loss_info.reduction_type == LossReductionType::Undefined)
    return false;
 
  // If there are invalid combination, add more condition-check here
  return true;
}

References onert::ir::train::OptimizerInfo::learning_rate, onert::ir::train::LossInfo::loss_code, onert::ir::train::OptimizerInfo::optim_code, onert::ir::train::LossInfo::reduction_type, and onert::ir::train::Undefined.

lossInfo()

const LossInfo & onert::ir::train::TrainingInfo::lossInfo ( ) const

inline

Definition at line 47 of file TrainingInfo.h.

{ return _loss_info; }

Referenced by onert::compiler::train::pass::LossInsertionPass::run().

operator=() [1/2]

TrainingInfo & onert::ir::train::TrainingInfo::operator= ( const TrainingInfo &  )

default

operator=() [2/2]

TrainingInfo & onert::ir::train::TrainingInfo::operator= ( TrainingInfo &&  )

default

optimizerInfo()

const OptimizerInfo & onert::ir::train::TrainingInfo::optimizerInfo ( ) const

inline

Definition at line 48 of file TrainingInfo.h.

{ return _optimizer_info; }

setBatchSize()

void onert::ir::train::TrainingInfo::setBatchSize ( const uint32_t  batch_size )

inline

Definition at line 55 of file TrainingInfo.h.

{ _batch_size = batch_size; }

setLossInfo()

void onert::ir::train::TrainingInfo::setLossInfo ( const LossInfo &  loss_info )

inline

Definition at line 56 of file TrainingInfo.h.

{ _loss_info = loss_info; }

setOptimizerInfo()

void onert::ir::train::TrainingInfo::setOptimizerInfo ( const OptimizerInfo &  optimizer_info )

inline

Definition at line 57 of file TrainingInfo.h.

{ _optimizer_info = optimizer_info; }

setTrainableOps()

void onert::ir::train::TrainingInfo::setTrainableOps ( const std::set< OperationIndex > &  trainable_ops )

inline

Definition at line 59 of file TrainingInfo.h.

  {
    _trainable_ops = trainable_ops;
  }

setVersion()

void onert::ir::train::TrainingInfo::setVersion ( const uint32_t  version )

inline

Definition at line 54 of file TrainingInfo.h.

{ _version = version; }

References version().

trainingStep() [1/2]

uint32_t & onert::ir::train::TrainingInfo::trainingStep ( )

inline

Definition at line 58 of file TrainingInfo.h.

{ return _training_step; }

trainingStep() [2/2]

const uint32_t & onert::ir::train::TrainingInfo::trainingStep ( ) const

inline

Definition at line 50 of file TrainingInfo.h.

{ return _training_step; }

version()

uint32_t onert::ir::train::TrainingInfo::version ( ) const

inline

Definition at line 46 of file TrainingInfo.h.

{ return _version; }

Referenced by setVersion().

---

The documentation for this class was generated from the following files:

• runtime/onert/core/include/ir/train/TrainingInfo.h
• runtime/onert/core/src/ir/train/TrainingInfo.cc