onert::ir::train::operation::Softmax Class Reference

#include <Softmax.h>

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

Public Member Functions
	Softmax (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::Softmax
	Softmax (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, const Param &param)
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
virtual std::string	name () const
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::Softmax
enum	Input { INPUT = 0 }

Detailed Description

Definition at line 26 of file Softmax.h.

Constructor & Destructor Documentation

Softmax()

onert::ir::train::operation::Softmax::Softmax

(

const OperationType &

operation

)

Definition at line 34 of file Softmax.cc.

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

Member Function Documentation

accept() [1/2]

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

overridevirtual

Reimplemented from onert::ir::operation::Softmax.

Definition at line 30 of file Softmax.cc.

{ v.visit(*this); }

accept() [2/2]

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

overridevirtual

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

Definition at line 32 of file Softmax.cc.

{ v.visit(*this); }

clone()

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

overridevirtual

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

Definition at line 25 of file Softmax.cc.

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

hasTrainableParameter()

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

inlineoverridevirtual

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

Definition at line 38 of file Softmax.h.

{ return false; }

---

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

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