ONE - On-device Neural Engine
|
#include <BinaryArithmetic.h>
Public Member Functions | |
BinaryArithmetic (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::BinaryArithmetic | |
BinaryArithmetic (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, const Param ¶m) | |
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::BinaryArithmetic | |
enum | Input { LHS = 0 , RHS } |
enum class | ArithmeticType { ADD , SUB , MUL , DIV } |
Definition at line 32 of file BinaryArithmetic.h.
onert::ir::train::operation::BinaryArithmetic::BinaryArithmetic | ( | const OperationType & | operation | ) |
Definition at line 40 of file BinaryArithmetic.cc.
|
overridevirtual |
Reimplemented from onert::ir::operation::BinaryArithmetic.
Definition at line 36 of file BinaryArithmetic.cc.
|
overridevirtual |
Implements onert::ir::train::ITrainableOperation.
Definition at line 38 of file BinaryArithmetic.cc.
|
overridevirtual |
Implements onert::ir::train::ITrainableOperation.
Definition at line 31 of file BinaryArithmetic.cc.
|
inlineoverridevirtual |
Implements onert::ir::train::ITrainableOperation.
Definition at line 44 of file BinaryArithmetic.h.