ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onert::ir::train::operation::BinaryArithmetic Class Reference

#include <BinaryArithmetic.h>

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

Public Member Functions

 BinaryArithmetic (const OperationType &operation)
 
std::unique_ptr< ITrainableOperationclone () 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::BinaryArithmetic
 BinaryArithmetic (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, const Param &param)
 
std::string name () const override
 
OpCode opcode () const final
 
const Paramparam () 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
 
Operationoperator= (const Operation &)=default
 
Operationoperator= (Operation &&)=default
 
virtual ~Operation ()
 
void replaceInputs (const OperandIndex &from, const OperandIndex &to) override
 
void replaceOutputs (const OperandIndex &from, const OperandIndex &to) override
 
OperandIndexSequencegetInputs ()
 
const OperandIndexSequencegetInputs () const override
 
const OperandIndexSequencegetOutputs () 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::BinaryArithmetic
enum  Input { LHS = 0 , RHS }
 
enum class  ArithmeticType { ADD , SUB , MUL , DIV }
 

Detailed Description

Definition at line 32 of file BinaryArithmetic.h.

Constructor & Destructor Documentation

◆ BinaryArithmetic()

onert::ir::train::operation::BinaryArithmetic::BinaryArithmetic ( const OperationType operation)

Definition at line 40 of file BinaryArithmetic.cc.

41 : OperationType{operation.getInputs(), operation.getOutputs(), operation.param()}
42{
43 // DO NOTHING
44}
OperationType

Member Function Documentation

◆ accept() [1/2]

void onert::ir::train::operation::BinaryArithmetic::accept ( OperationVisitor v) const
overridevirtual

Reimplemented from onert::ir::operation::BinaryArithmetic.

Definition at line 36 of file BinaryArithmetic.cc.

36{ v.visit(*this); }

◆ accept() [2/2]

void onert::ir::train::operation::BinaryArithmetic::accept ( TrainableOperationVisitor v) const
overridevirtual

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

Definition at line 38 of file BinaryArithmetic.cc.

38{ v.visit(*this); }

◆ clone()

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

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

Definition at line 31 of file BinaryArithmetic.cc.

32{
33 return std::make_unique<BinaryArithmetic>(*this);
34}

◆ hasTrainableParameter()

bool onert::ir::train::operation::BinaryArithmetic::hasTrainableParameter ( ) const
inlineoverridevirtual

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

Definition at line 44 of file BinaryArithmetic.h.

44{ return false; }

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