#include <ElementwiseUnary.h>

Collaboration diagram for onert::ir::operation::ElementwiseUnary:

[legend]

Data Structures
struct	Param

Public Types
enum	Input { INPUT = 0 }

enum class	Type { ABS , CAST , COS , DEQUANTIZE , ERF , EXP , FLOOR , LOG , LOGICAL_NOT , NEG , QUANTIZE , ROUND , RSQRT , SIN , SQRT , SQUARE , ZEROS_LIKE }

Public Member Functions
	ElementwiseUnary (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, const Param &param)

void	accept (OperationVisitor &v) const override

std::string	name () 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

const OperandIndexSequence &	getOutputs () const override

void	setInputs (const OperandIndexSequence &indexes)

void	setOutputs (const OperandIndexSequence &indexes)

Public Member Functions inherited from onert::ir::IOperation
virtual	~IOperation ()=default

Detailed Description

Definition at line 29 of file ElementwiseUnary.h.

Member Enumeration Documentation

◆ Input

enum onert::ir::operation::ElementwiseUnary::Input

Enumerator
INPUT

Definition at line 32 of file ElementwiseUnary.h.

  {
    INPUT = 0
  };

◆ Type

enum class onert::ir::operation::ElementwiseUnary::Type

strong

Enumerator
ABS
CAST
COS
DEQUANTIZE
ERF
EXP
FLOOR
LOG
LOGICAL_NOT
NEG
QUANTIZE
ROUND
RSQRT
SIN
SQRT
SQUARE
ZEROS_LIKE

Definition at line 37 of file ElementwiseUnary.h.

  {
    ABS,
    CAST,
    COS,
    DEQUANTIZE,
    ERF,
    EXP,
    FLOOR,
    LOG,
    LOGICAL_NOT,
    NEG,
    QUANTIZE,
    ROUND,
    RSQRT,
    SIN,
    SQRT,
    SQUARE,
    ZEROS_LIKE
  };

Constructor & Destructor Documentation

◆ ElementwiseUnary()

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

Definition at line 31 of file ElementwiseUnary.cc.

  : Operation{OperandConstraint::createExact(1u), inputs, outputs,
              OperandConstraint::createExact(1u)},
    _param{param}
{
}

Member Function Documentation

◆ accept()

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

overridevirtual

Implements onert::ir::IOperation.

Definition at line 29 of file ElementwiseUnary.cc.

29{ v.visit(*this); }

◆ name()

std::string onert::ir::operation::ElementwiseUnary::name ( ) const

overridevirtual

Reimplemented from onert::ir::IOperation.

Definition at line 39 of file ElementwiseUnary.cc.

{
  using ElementwiseUnaryType = onert::ir::operation::ElementwiseUnary::Type;
  static const std::unordered_map<ElementwiseUnaryType, std::string> name_map{
    {ElementwiseUnaryType::ABS, std::string{"Abs"}},
    {ElementwiseUnaryType::CAST, std::string{"Cast"}},
    {ElementwiseUnaryType::COS, std::string{"Cos"}},
    {ElementwiseUnaryType::DEQUANTIZE, std::string{"Dequantize"}},
    {ElementwiseUnaryType::ERF, std::string{"Erf"}},
    {ElementwiseUnaryType::EXP, std::string{"Exp"}},
    {ElementwiseUnaryType::FLOOR, std::string{"Floor"}},
    {ElementwiseUnaryType::LOG, std::string{"Log"}},
    {ElementwiseUnaryType::LOGICAL_NOT, std::string{"LogicalNot"}},
    {ElementwiseUnaryType::NEG, std::string{"Neg"}},
    {ElementwiseUnaryType::QUANTIZE, std::string{"Quantize"}},
    {ElementwiseUnaryType::ROUND, std::string{"Round"}},
    {ElementwiseUnaryType::RSQRT, std::string{"RSqrt"}},
    {ElementwiseUnaryType::SIN, std::string{"Sin"}},
    {ElementwiseUnaryType::SQRT, std::string{"Sqrt"}},
    {ElementwiseUnaryType::SQUARE, std::string{"Square"}},
    {ElementwiseUnaryType::ZEROS_LIKE, std::string{"ZerosLike"}}};
  return name_map.at(_param.op_type);
}

References onert::ir::operation::ElementwiseUnary::Param::op_type.

◆ opcode()

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

inlinefinalvirtual

Implements onert::ir::IOperation.

Definition at line 70 of file ElementwiseUnary.h.

70{ return OpCode::ElementwiseUnary; }

◆ param()

const Param & onert::ir::operation::ElementwiseUnary::param ( ) const

inline

Definition at line 73 of file ElementwiseUnary.h.

73{ return _param; }

Referenced by onert::backend::cpu::KernelGenerator::visit(), and onert::ir::OperationValidator::visit().

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

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

Data Structures

Public Types

Public Member Functions