ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Data Structures | Public Types | Public Member Functions
onert::ir::operation::RmsNorm Class Reference

#include <RmsNorm.h>

Collaboration diagram for onert::ir::operation::RmsNorm:
Collaboration graph
[legend]

Data Structures

struct  Param
 

Public Types

enum  Input { INPUT = 0 , GAMMA }
 

Public Member Functions

 RmsNorm (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, const Param &param)
 
void accept (OperationVisitor &v) 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
 
OperandIndexSequence getUsedInputSet () const override
 
const OperandIndexSequence & getOutputs () const override
 
OperandIndexSequence getUsedOutputSet () 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 26 of file RmsNorm.h.

Member Enumeration Documentation

◆ Input

enum onert::ir::operation::RmsNorm::Input
Enumerator
INPUT 
GAMMA 

Definition at line 29 of file RmsNorm.h.

30 {
31 INPUT = 0,
32 GAMMA
33 };

Constructor & Destructor Documentation

◆ RmsNorm()

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

Definition at line 25 of file RmsNorm.cc.

27 : Operation{OperandConstraint::createExact(2u), inputs, outputs}, _param{param}
28{
29}
onert::ir::OperandConstraint::createExact
static OperandConstraint createExact(uint32_t exact)
Definition OperandConstraint.h:34
onert::ir::operation::RmsNorm::param
const Param & param() const
Definition RmsNorm.h:49

Member Function Documentation

◆ accept()

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

Implements onert::ir::IOperation.

Definition at line 23 of file RmsNorm.cc.

23{ v.visit(*this); }

◆ opcode()

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

Implements onert::ir::IOperation.

Definition at line 46 of file RmsNorm.h.

46{ return OpCode::RmsNorm; }

◆ param()

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

Definition at line 49 of file RmsNorm.h.

49{ return _param; }
The documentation for this class was generated from the following files: