ONE - On-device Neural Engine
|
#include <Permute.h>
Public Member Functions | |
Permute (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::Permute | |
OpCode | opcode () const final |
Permute (const OperandIndex &input, const OperandIndex &output) | |
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 |
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 |
onert::ir::train::operation::Permute::Permute | ( | const OperationType & | operation | ) |
Definition at line 40 of file Permute.cc.
|
overridevirtual |
Reimplemented from onert::ir::operation::Permute.
Definition at line 36 of file Permute.cc.
|
overridevirtual |
Implements onert::ir::train::ITrainableOperation.
Definition at line 38 of file Permute.cc.
|
overridevirtual |
Implements onert::ir::train::ITrainableOperation.
Definition at line 31 of file Permute.cc.
|
inlineoverridevirtual |
Implements onert::ir::train::ITrainableOperation.
Definition at line 44 of file Permute.h.