onert::ir::train::operation::FullyConnected Class Reference

#include <FullyConnected.h>

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

Public Member Functions
	FullyConnected (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::FullyConnected
	FullyConnected (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::FullyConnected
enum	Input { INPUT = 0 , WEIGHT , BIAS }

Detailed Description

Definition at line 26 of file FullyConnected.h.

Constructor & Destructor Documentation

FullyConnected()

onert::ir::train::operation::FullyConnected::FullyConnected

(

const OperationType &

operation

)

Definition at line 34 of file FullyConnected.cc.

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

Member Function Documentation

accept() [1/2]

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

overridevirtual

Reimplemented from onert::ir::operation::FullyConnected.

Definition at line 30 of file FullyConnected.cc.

{ v.visit(*this); }

accept() [2/2]

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

overridevirtual

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

Definition at line 32 of file FullyConnected.cc.

{ v.visit(*this); }

clone()

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

overridevirtual

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

Definition at line 25 of file FullyConnected.cc.

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

hasTrainableParameter()

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

inlineoverridevirtual

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

Definition at line 38 of file FullyConnected.h.

{ return true; }

---

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

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