onert::exec::train::TrainableFnSequence Class Reference

#include <TrainableFnSequence.h>

Public Member Functions
void	forward (bool training)
void	backward (uint32_t training_step, bool weight_update_enabled)
void	append (std::unique_ptr< ITrainableFunction > &&fn)
void	append (std::unique_ptr< IGradientApplier > &&applier)
void	iterate (const std::function< void(ITrainableFunction &)> &fn)

Data Fields
std::vector< std::unique_ptr< ITrainableFunction > >	_functions
std::vector< std::unique_ptr< IGradientApplier > >	_appliers

Detailed Description

Definition at line 33 of file TrainableFnSequence.h.

Member Function Documentation

append() [1/2]

void onert::exec::train::TrainableFnSequence::append

(

std::unique_ptr< IGradientApplier > &&

applier

)

Definition at line 54 of file TrainableFnSequence.cc.

{
  _appliers.push_back(std::move(applier));
}

References _appliers.

append() [2/2]

void onert::exec::train::TrainableFnSequence::append

(

std::unique_ptr< ITrainableFunction > &&

fn

)

Definition at line 49 of file TrainableFnSequence.cc.

{
  _functions.push_back(std::move(function));
}

References _functions.

backward()

void onert::exec::train::TrainableFnSequence::backward	(	uint32_t	training_step,
		bool	weight_update_enabled
	)

Definition at line 34 of file TrainableFnSequence.cc.

{
  for (auto it = _functions.rbegin(); it != _functions.rend(); ++it)
  {
    (*it)->backward();
  }
  if (weight_update_enabled)
  {
    for (const auto &applier : _appliers)
    {
      applier->applyGradient(training_step);
    }
  }
}

References _appliers, and _functions.

forward()

void onert::exec::train::TrainableFnSequence::forward

(

bool

training

)

Definition at line 26 of file TrainableFnSequence.cc.

{
  for (const auto &function : _functions)
  {
    function->forward(training);
  }
}

References _functions.

iterate()

void onert::exec::train::TrainableFnSequence::iterate

(

const std::function< void(ITrainableFunction &)> &

fn

)

Definition at line 59 of file TrainableFnSequence.cc.

{
  for (const auto &func : _functions)
  {
    fn(*func);
  }
}

References _functions.

Field Documentation

_appliers

std::vector<std::unique_ptr<IGradientApplier> > onert::exec::train::TrainableFnSequence::_appliers

Definition at line 46 of file TrainableFnSequence.h.

Referenced by append(), and backward().

_functions

std::vector<std::unique_ptr<ITrainableFunction> > onert::exec::train::TrainableFnSequence::_functions

Definition at line 45 of file TrainableFnSequence.h.

Referenced by append(), backward(), forward(), and iterate().

---

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

• runtime/onert/core/include/exec/train/TrainableFnSequence.h
• runtime/onert/core/src/exec/train/TrainableFnSequence.cc