onert::ir::train::operation::Reduce Class Reference

#include <Reduce.h>

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

Public Member Functions
	Reduce (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::Reduce
	Reduce (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
const OperandIndexSequence &	getOutputs () 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::Reduce
enum	Input { INPUT = 0 , AXES = 1 }
enum class	ReduceType {   ALL , ANY , MAX , MEAN ,   MIN , PROD , SUM }

Detailed Description

Definition at line 32 of file Reduce.h.

Constructor & Destructor Documentation

Reduce()

onert::ir::train::operation::Reduce::Reduce

(

const OperationType &

operation

)

Definition at line 40 of file Reduce.cc.

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

Member Function Documentation

accept() [1/2]

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

overridevirtual

Reimplemented from onert::ir::operation::Reduce.

Definition at line 36 of file Reduce.cc.

{ v.visit(*this); }

accept() [2/2]

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

overridevirtual

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

Definition at line 38 of file Reduce.cc.

{ v.visit(*this); }

clone()

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

overridevirtual

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

Definition at line 31 of file Reduce.cc.

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

hasTrainableParameter()

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

inlineoverridevirtual

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

Definition at line 44 of file Reduce.h.

{ return false; }

---

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

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