onert::backend::cpu::ops::RmsNormLayer Class Reference

#include <RmsNormLayer.h>

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

Public Member Functions
	RmsNormLayer ()
void	configure (const IPortableTensor *input, const IPortableTensor *gamma, float epsilon, IPortableTensor *output)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 32 of file RmsNormLayer.h.

Constructor & Destructor Documentation

RmsNormLayer()

onert::backend::cpu::ops::RmsNormLayer::RmsNormLayer ( )

inline

Definition at line 35 of file RmsNormLayer.h.

                 : _input(nullptr), _output(nullptr), _gamma(nullptr), _epsilon(1e-06f)
  {
    // Nothing
  }

Member Function Documentation

configure()

void onert::backend::cpu::ops::RmsNormLayer::configure	(	const IPortableTensor *	input,
		const IPortableTensor *	gamma,
		float	epsilon,
		IPortableTensor *	output
	)

Definition at line 33 of file RmsNormLayer.cc.

{
  assert(input != nullptr);
  assert(output != nullptr);
 
  _input = input;
  _output = output;
  _gamma = gamma;
  _epsilon = epsilon;
}

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 45 of file RmsNormLayer.cc.

{
  switch (_input->data_type())
  {
    case OperandType::FLOAT32:
      nnfw::cker::RmsNormParams param;
      param.epsilon = _epsilon;
      nnfw::cker::RmsNorm(param, getShape(_input), getBuffer<float>(_input), getShape(_gamma),
                          getBuffer<float>(_gamma), getShape(_output), getBuffer<float>(_output));
      break;
 
    default:
      throw std::runtime_error{"RmsNorm: Unsupported data type"};
  }
}

References onert::backend::IPortableTensor::data_type(), nnfw::cker::RmsNormParams::epsilon, onert::backend::cpu::ops::getShape(), and nnfw::cker::RmsNorm().

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

---

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

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