onert::ir::train::ITrainableOperation Class Reference

#include <ITrainableOperation.h>

Collaboration diagram for onert::ir::train::ITrainableOperation:

Public Member Functions
virtual	~ITrainableOperation ()=default
virtual std::unique_ptr< ITrainableOperation >	clone () const =0
virtual void	accept (OperationVisitor &v) const override=0
virtual void	accept (TrainableOperationVisitor &v) const =0
virtual bool	hasTrainableParameter () const =0
virtual void	disableWeightsUpdate ()=0
virtual void	enableWeightsUpdate ()=0
virtual bool	isWeightsUpdateEnabled () const =0
virtual void	enableBackward ()=0
virtual void	disableBackward ()=0
virtual bool	isRequiredForBackward () const =0
Public Member Functions inherited from onert::ir::IOperation
virtual	~IOperation ()=default
virtual std::string	name () const
virtual OpCode	opcode () const =0
virtual void	replaceInputs (const OperandIndex &from, const OperandIndex &to)=0
virtual void	replaceOutputs (const OperandIndex &from, const OperandIndex &to)=0
virtual const OperandIndexSequence &	getInputs () const =0
virtual const OperandIndexSequence &	getOutputs () const =0

Detailed Description

Definition at line 33 of file ITrainableOperation.h.

Constructor & Destructor Documentation

~ITrainableOperation()

virtual onert::ir::train::ITrainableOperation::~ITrainableOperation ( )

virtualdefault

Member Function Documentation

accept() [1/2]

virtual void onert::ir::train::ITrainableOperation::accept ( OperationVisitor &  v ) const

overridepure virtual

Implements onert::ir::IOperation.

Implemented in onert::ir::train::operation::BinaryArithmetic, onert::ir::train::operation::Conv2D, onert::ir::train::operation::DepthwiseConv2D, onert::ir::train::operation::ElementwiseActivation, onert::ir::train::operation::FullyConnected, onert::ir::train::operation::Loss, onert::ir::train::operation::Pad, onert::ir::train::operation::Permute, onert::ir::train::operation::Pool2D, onert::ir::train::operation::Reduce, onert::ir::train::operation::Reshape, onert::ir::train::operation::Softmax, and onert::ir::train::operation::UntrainableOperation< OperationType, typename >.

Referenced by onert::backend::builtin::train::KernelGenerator::generate().

accept() [2/2]

virtual void onert::ir::train::ITrainableOperation::accept ( TrainableOperationVisitor &  v ) const

pure virtual

Implemented in onert::ir::train::operation::UntrainableOperation< OperationType, typename >, onert::ir::train::operation::BinaryArithmetic, onert::ir::train::operation::Conv2D, onert::ir::train::operation::DepthwiseConv2D, onert::ir::train::operation::ElementwiseActivation, onert::ir::train::operation::FullyConnected, onert::ir::train::operation::Loss, onert::ir::train::operation::Pad, onert::ir::train::operation::Permute, onert::ir::train::operation::Pool2D, onert::ir::train::operation::Reduce, onert::ir::train::operation::Reshape, and onert::ir::train::operation::Softmax.

clone()

virtual std::unique_ptr< ITrainableOperation > onert::ir::train::ITrainableOperation::clone ( ) const

pure virtual

Implemented in onert::ir::train::operation::BinaryArithmetic, onert::ir::train::operation::Conv2D, onert::ir::train::operation::DepthwiseConv2D, onert::ir::train::operation::ElementwiseActivation, onert::ir::train::operation::FullyConnected, onert::ir::train::operation::Loss, onert::ir::train::operation::Pad, onert::ir::train::operation::Permute, onert::ir::train::operation::Pool2D, onert::ir::train::operation::Reduce, onert::ir::train::operation::Reshape, onert::ir::train::operation::Softmax, and onert::ir::train::operation::UntrainableOperation< OperationType, typename >.

disableBackward()

virtual void onert::ir::train::ITrainableOperation::disableBackward ( )

pure virtual

Implemented in onert::ir::train::TrainableOperation.

Referenced by onert::ir::train::TrainableGraph::disableBackward().

disableWeightsUpdate()

virtual void onert::ir::train::ITrainableOperation::disableWeightsUpdate ( )

pure virtual

Implemented in onert::ir::train::TrainableOperation.

enableBackward()

virtual void onert::ir::train::ITrainableOperation::enableBackward ( )

pure virtual

Implemented in onert::ir::train::TrainableOperation.

enableWeightsUpdate()

virtual void onert::ir::train::ITrainableOperation::enableWeightsUpdate ( )

pure virtual

Implemented in onert::ir::train::TrainableOperation.

hasTrainableParameter()

virtual bool onert::ir::train::ITrainableOperation::hasTrainableParameter ( ) const

pure virtual

Implemented in onert::ir::train::operation::BinaryArithmetic, onert::ir::train::operation::Conv2D, onert::ir::train::operation::DepthwiseConv2D, onert::ir::train::operation::ElementwiseActivation, onert::ir::train::operation::FullyConnected, onert::ir::train::operation::Loss, onert::ir::train::operation::Pad, onert::ir::train::operation::Permute, onert::ir::train::operation::Pool2D, onert::ir::train::operation::Reduce, onert::ir::train::operation::Reshape, onert::ir::train::operation::Softmax, and onert::ir::train::operation::UntrainableOperation< OperationType, typename >.

isRequiredForBackward()

virtual bool onert::ir::train::ITrainableOperation::isRequiredForBackward ( ) const

pure virtual

Implemented in onert::ir::train::TrainableOperation.

Referenced by onert::ir::train::TrainableGraph::essentialBackwardOrder().

isWeightsUpdateEnabled()

virtual bool onert::ir::train::ITrainableOperation::isWeightsUpdateEnabled ( ) const

pure virtual

Implemented in onert::ir::train::TrainableOperation.

---

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

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