#include <Attention.h>

Collaboration diagram for onert::ir::operation::Attention:

[legend]

Public Types
enum	Input { INPUT = 0 , WQ = 1 , WK = 2 , WV = 3 , WO = 4 , COS = 5 , SIN = 6 , MASK = 7 , K_CACHE = 8 , V_CACHE = 9 , POS = 10 }

Public Member Functions
	Attention (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs)

void	accept (OperationVisitor &v) const override

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

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 Attention.h.

Member Enumeration Documentation

◆ Input

enum onert::ir::operation::Attention::Input

Enumerator
INPUT
WQ
WK
WV
WO
COS
SIN
MASK
K_CACHE
V_CACHE
POS

Definition at line 29 of file Attention.h.

  {
    INPUT = 0,
    WQ = 1,
    WK = 2,
    WV = 3,
    WO = 4,
    COS = 5,
    SIN = 6,
    MASK = 7,
    K_CACHE = 8,
    V_CACHE = 9,
    POS = 10,
  };

Constructor & Destructor Documentation

◆ Attention()

onert::ir::operation::Attention::Attention	(	const OperandIndexSequence &	inputs,
		const OperandIndexSequence &	outputs
	)

Definition at line 25 of file Attention.cc.

  : Operation{OperandConstraint::createExact(11u), inputs, outputs}
{
}

Member Function Documentation

◆ accept()

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

overridevirtual

Implements onert::ir::IOperation.

Definition at line 23 of file Attention.cc.

23{ v.visit(*this); }

◆ opcode()

OpCode onert::ir::operation::Attention::opcode ( ) const

inlinefinalvirtual

Implements onert::ir::IOperation.

Definition at line 47 of file Attention.h.

47{ return OpCode::Attention; }

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

runtime/onert/core/include/ir/operation/Attention.h
runtime/onert/core/src/ir/operation/Attention.cc

Public Types

Public Member Functions