onert::ir::train::operation::ElementwiseActivation Class Reference

#include <ElementwiseActivation.h>

Collaboration diagram for onert::ir::train::operation::ElementwiseActivation:

Public Member Functions
	ElementwiseActivation (const OperationType &operation)
std::unique_ptr< ITrainableOperation >	clone () const override
void	accept (OperationVisitor &v) const override
void	accept (TrainableOperationVisitor &v) const override
bool	hasTrainableParameter () const override
Public Member Functions inherited from onert::ir::operation::ElementwiseActivation
	ElementwiseActivation (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, const Param &param)
std::string	name () const override
OpCode	opcode () const final
const Param &	param () const
Public Member Functions inherited from onert::ir::Operation
	Operation (OperandConstraint input_constr, const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, OperandConstraint output_constr=OperandConstraint::createAny())
	Operation (OperandConstraint input_constr, OperandConstraint output_constr=OperandConstraint::createAny())
	Operation (const Operation &)=default
	Operation (Operation &&)=default
Operation &	operator= (const Operation &)=default
Operation &	operator= (Operation &&)=default
virtual	~Operation ()
void	replaceInputs (const OperandIndex &from, const OperandIndex &to) override
void	replaceOutputs (const OperandIndex &from, const OperandIndex &to) override
OperandIndexSequence &	getInputs ()
const OperandIndexSequence &	getInputs () const override
OperandIndexSequence	getUsedInputSet () const override
const OperandIndexSequence &	getOutputs () const override
OperandIndexSequence	getUsedOutputSet () const override
void	setInputs (const OperandIndexSequence &indexes)
void	setOutputs (const OperandIndexSequence &indexes)
Public Member Functions inherited from onert::ir::IOperation
virtual	~IOperation ()=default
Public Member Functions inherited from onert::ir::train::TrainableOperation
virtual	~TrainableOperation ()=default
void	disableWeightsUpdate () final
void	enableWeightsUpdate () final
virtual bool	isWeightsUpdateEnabled () const final
void	enableBackward () final
void	disableBackward () final
virtual bool	isRequiredForBackward () const final
Public Member Functions inherited from onert::ir::train::ITrainableOperation
virtual	~ITrainableOperation ()=default

Additional Inherited Members
Public Types inherited from onert::ir::operation::ElementwiseActivation
enum	Input { INPUT = 0 }
enum class	Type {   ELU , LOGISTIC , RELU , TANH ,   LEAKY_RELU , GELU }
Static Public Attributes inherited from onert::ir::operation::ElementwiseActivation
static float	infinity = std::numeric_limits<float>::infinity()

Detailed Description

Definition at line 26 of file ElementwiseActivation.h.

Constructor & Destructor Documentation

ElementwiseActivation()

onert::ir::train::operation::ElementwiseActivation::ElementwiseActivation

(

const OperationType &

operation

)

Definition at line 34 of file ElementwiseActivation.cc.

  : OperationType{operation.getInputs(), operation.getOutputs(), operation.param()}
{
  // DO NOTHING
}

Member Function Documentation

accept() [1/2]

void onert::ir::train::operation::ElementwiseActivation::accept ( OperationVisitor &  v ) const

overridevirtual

Reimplemented from onert::ir::operation::ElementwiseActivation.

Definition at line 30 of file ElementwiseActivation.cc.

{ v.visit(*this); }

accept() [2/2]

void onert::ir::train::operation::ElementwiseActivation::accept ( TrainableOperationVisitor &  v ) const

overridevirtual

Implements onert::ir::train::ITrainableOperation.

Definition at line 32 of file ElementwiseActivation.cc.

{ v.visit(*this); }

clone()

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

overridevirtual

Implements onert::ir::train::ITrainableOperation.

Definition at line 25 of file ElementwiseActivation.cc.

{
  return std::make_unique<ElementwiseActivation>(*this);
}

hasTrainableParameter()

bool onert::ir::train::operation::ElementwiseActivation::hasTrainableParameter ( ) const

inlineoverridevirtual

Implements onert::ir::train::ITrainableOperation.

Definition at line 38 of file ElementwiseActivation.h.

{ return false; }

---

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

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