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

#include <RNN.h>

Collaboration diagram for onert::ir::operation::RNN:

Data Structures

struct  Param
 

Public Types

enum  Input {
  INPUT = 0 , WEIGHTS = 1 , RECURRENT_WEIGHTS = 2 , BIAS = 3 ,
  HIDDEN_STATE_IN = 4
}
 
enum  Output { OUTPUT = 0 , HIDDEN_STATE_OUT = 1 }
 

Public Member Functions

 RNN (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, const Param &param)
 
void accept (OperationVisitor &v) 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
 
virtual std::string name () const
 

Detailed Description

Definition at line 29 of file RNN.h.

Member Enumeration Documentation

◆ Input

Enumerator
INPUT 
WEIGHTS 
RECURRENT_WEIGHTS 
BIAS 
HIDDEN_STATE_IN 

Definition at line 32 of file RNN.h.

◆ Output

Enumerator
OUTPUT 
HIDDEN_STATE_OUT 

Definition at line 41 of file RNN.h.

42 {
43 OUTPUT = 0,
45 };

Constructor & Destructor Documentation

◆ RNN()

onert::ir::operation::RNN::RNN ( const OperandIndexSequence inputs,
const OperandIndexSequence outputs,
const Param param 
)

Definition at line 29 of file RNN.cc.

32{
33}
static OperandConstraint createExact(uint32_t exact)
const Param & param() const
Definition RNN.h:60

Member Function Documentation

◆ accept()

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

Implements onert::ir::IOperation.

Definition at line 27 of file RNN.cc.

27{ v.visit(*this); }

◆ opcode()

OpCode onert::ir::operation::RNN::opcode ( ) const
inlinefinalvirtual

Implements onert::ir::IOperation.

Definition at line 57 of file RNN.h.

57{ return OpCode::RNN; }

◆ param()

const Param & onert::ir::operation::RNN::param ( ) const
inline

Definition at line 60 of file RNN.h.

60{ return _param; }

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