onert::backend::cpu::ops::RoPELayer Class Reference

#include <RoPELayer.h>

Collaboration diagram for onert::backend::cpu::ops::RoPELayer:

Public Member Functions
	RoPELayer ()
	~RoPELayer ()
void	configure (const IPortableTensor *input, const IPortableTensor *sin, const IPortableTensor *cos, const nnfw::cker::RoPEMode mode, IPortableTensor *output)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 34 of file RoPELayer.h.

Constructor & Destructor Documentation

RoPELayer()

onert::backend::cpu::ops::RoPELayer::RoPELayer

(

)

Definition at line 30 of file RoPELayer.cc.

  : _input(nullptr), _sin(nullptr), _cos(nullptr), _mode(nnfw::cker::RoPEMode::kGptNeox),
    _output(nullptr)
{
  // DO NOTHING
}

~RoPELayer()

onert::backend::cpu::ops::RoPELayer::~RoPELayer ( )

default

Member Function Documentation

configure()

void onert::backend::cpu::ops::RoPELayer::configure	(	const IPortableTensor *	input,
		const IPortableTensor *	sin,
		const IPortableTensor *	cos,
		const nnfw::cker::RoPEMode	mode,
		IPortableTensor *	output
	)

Definition at line 39 of file RoPELayer.cc.

{
  assert(input != nullptr);
  assert(sin != nullptr);
  assert(cos != nullptr);
  assert(output != nullptr);
 
  _input = input;
  _sin = sin;
  _cos = cos;
  _mode = mode;
  _output = output;
}

run()

void onert::backend::cpu::ops::RoPELayer::run ( )

overridevirtual

Implements onert::exec::IFunction.

Definition at line 65 of file RoPELayer.cc.

{
  switch (_input->data_type())
  {
    case OperandType::FLOAT32:
      rope<float>();
      break;
    case OperandType::INT32:
      rope<int32_t>();
      break;
    case OperandType::INT64:
      rope<int64_t>();
      break;
    case OperandType::QUANT_UINT8_ASYMM:
      rope<uint8_t>();
      break;
    case OperandType::QUANT_INT8_ASYMM:
      rope<int8_t>();
      break;
    default:
      throw std::runtime_error("RoPE: unsupported data type");
  }
}

References onert::backend::IPortableTensor::data_type().

Referenced by package.infer.session::inference().

---

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

• runtime/onert/backend/cpu/ops/RoPELayer.h
• runtime/onert/backend/cpu/ops/RoPELayer.cc