onert::exec::train::ITrainableFunction Class Reference

#include <ITrainableFunction.h>

Public Member Functions
virtual	~ITrainableFunction ()=default
virtual void	forward (bool training)=0
virtual void	backward ()=0
virtual std::optional< backend::train::LayerScopeTensors >	registerLayerScopeTensors ()

Detailed Description

Definition at line 28 of file ITrainableFunction.h.

Constructor & Destructor Documentation

~ITrainableFunction()

virtual onert::exec::train::ITrainableFunction::~ITrainableFunction ( )

virtualdefault

Member Function Documentation

backward()

virtual void onert::exec::train::ITrainableFunction::backward ( )

pure virtual

Implemented in onert::backend::train::ops::BackPropAccumulator, onert::backend::train::ops::BackPropInitializer, onert::backend::train::ops::BinaryArithmeticLayer, onert::backend::train::ops::ConvolutionLayer, onert::backend::train::ops::DepthwiseConvolutionLayer, onert::backend::train::ops::ElementwiseActivationLayer, onert::backend::train::ops::FullyConnectedLayer, onert::backend::train::ops::LossCategoricalCrossentropyLayer, onert::backend::train::ops::LossMeanSquaredErrorLayer, onert::backend::train::ops::MeanLayer, onert::backend::train::ops::PadLayer, onert::backend::train::ops::PoolLayer, onert::backend::train::ops::ReshapeLayer, onert::backend::train::ops::SoftMaxLayer, and onert::backend::builtin::train::kernel::PermuteLayer.

forward()

virtual void onert::exec::train::ITrainableFunction::forward ( bool  training )

pure virtual

Implemented in onert::backend::train::ops::BackPropAccumulator, onert::backend::train::ops::BackPropInitializer, onert::backend::train::ops::BinaryArithmeticLayer, onert::backend::train::ops::ConvolutionLayer, onert::backend::train::ops::DepthwiseConvolutionLayer, onert::backend::train::ops::ElementwiseActivationLayer, onert::backend::train::ops::FullyConnectedLayer, onert::backend::train::ops::LossCategoricalCrossentropyLayer, onert::backend::train::ops::LossMeanSquaredErrorLayer, onert::backend::train::ops::MeanLayer, onert::backend::train::ops::PadLayer, onert::backend::train::ops::PoolLayer, onert::backend::train::ops::ReshapeLayer, onert::backend::train::ops::SoftMaxLayer, and onert::backend::builtin::train::kernel::PermuteLayer.

registerLayerScopeTensors()

virtual std::optional< backend::train::LayerScopeTensors > onert::exec::train::ITrainableFunction::registerLayerScopeTensors ( )

inlinevirtual

Definition at line 36 of file ITrainableFunction.h.

  {
    return std::nullopt;
  }

---

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

• runtime/onert/core/include/exec/train/ITrainableFunction.h