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

#include <Permute.h>

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

Public Member Functions

 Permute (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::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
 
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
 
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
 

Detailed Description

Definition at line 32 of file Permute.h.

Constructor & Destructor Documentation

◆ Permute()

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

Definition at line 40 of file Permute.cc.

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

Member Function Documentation

◆ accept() [1/2]

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

Reimplemented from onert::ir::operation::Permute.

Definition at line 36 of file Permute.cc.

36{ v.visit(*this); }

◆ accept() [2/2]

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

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

Definition at line 38 of file Permute.cc.

38{ v.visit(*this); }

◆ clone()

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

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

Definition at line 31 of file Permute.cc.

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

◆ hasTrainableParameter()

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

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

Definition at line 44 of file Permute.h.

44{ return false; }

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