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

#include <Pad.h>

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

Public Member Functions

 Pad (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::Pad
 Pad (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs)
 
OpCode opcode () const final
 
- 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
 

Additional Inherited Members

- Public Types inherited from onert::ir::operation::Pad
enum  Input { INPUT = 0 , PAD = 1 , VALUE = 2 }
 

Detailed Description

Definition at line 32 of file Pad.h.

Constructor & Destructor Documentation

◆ Pad()

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

Definition at line 37 of file Pad.cc.

38 : OperationType{operation.getInputs(), operation.getOutputs()}
39{
40 // DO NOTHING
41}
OperationType

Member Function Documentation

◆ accept() [1/2]

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

Reimplemented from onert::ir::operation::Pad.

Definition at line 33 of file Pad.cc.

33{ v.visit(*this); }

◆ accept() [2/2]

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

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

Definition at line 35 of file Pad.cc.

35{ v.visit(*this); }

◆ clone()

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

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

Definition at line 31 of file Pad.cc.

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

◆ hasTrainableParameter()

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

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

Definition at line 44 of file Pad.h.

44{ return false; }

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