#include <MeanLayer.h>

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

[legend]

Public Member Functions
	MeanLayer ()

void	MeanFloat32 ()

void	MeanQuant8 ()

void	configure (const IPortableTensor input, const IPortableTensor axes, IPortableTensor *output, bool keep_dims)

void	run () override

Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default

virtual void	prepare ()

Protected Attributes
const IPortableTensor *	_input

const IPortableTensor *	_axes

IPortableTensor *	_output

bool	_keep_dims

Detailed Description

Definition at line 33 of file MeanLayer.h.

Constructor & Destructor Documentation

◆ MeanLayer()

onert::backend::cpu::ops::MeanLayer::MeanLayer ( )

Definition at line 32 of file MeanLayer.cc.

                     : _input(nullptr), _axes(nullptr), _output(nullptr), _keep_dims(false)
{
  // DO NOTHING
}

Member Function Documentation

◆ configure()

void onert::backend::cpu::ops::MeanLayer::configure	(	const IPortableTensor *	input,
		const IPortableTensor *	axes,
		IPortableTensor *	output,
		bool	keep_dims
	)

Definition at line 64 of file MeanLayer.cc.

{
  _input = input;
  _axes = axes;
  _output = output;
  _keep_dims = keep_dims;
 
  if (_input->data_type() != OperandType::FLOAT32 &&
      _input->data_type() != OperandType::QUANT_UINT8_ASYMM)
    throw std::runtime_error{"Mean: unsupported data type"};
}

References _axes, _input, _keep_dims, _output, and onert::backend::IPortableTensor::data_type().

◆ MeanFloat32()

void onert::backend::cpu::ops::MeanLayer::MeanFloat32 ( )

Definition at line 37 of file MeanLayer.cc.

{
  const auto inputShape = getShape(_input);
  const auto axisVec = getReducerAxes(_axes);
  bool axis_is_1_and_2 =
    _keep_dims && inputShape.DimensionsCount() == 4 && axisVec.size() == 2 &&
    ((axisVec[0] == 1 && axisVec[1] == 2) || (axisVec[0] == 2 && axisVec[1] == 1));
 
  if (axis_is_1_and_2)
  {
    nnfw::cker::MeanAxis1And2(inputShape, getBuffer<float>(_input), getShape(_output),
                              getBuffer<float>(_output));
  }
  else
  {
    nnfw::cker::Mean(inputShape, getBuffer<float>(_input), getShape(_output),
                     getBuffer<float>(_output), axisVec);
  }
}

References _axes, _input, _keep_dims, _output, onert::backend::cpu::ops::getReducerAxes(), onert::backend::cpu::ops::getShape(), nnfw::cker::Mean(), and nnfw::cker::MeanAxis1And2().

Referenced by run().

◆ MeanQuant8()

void onert::backend::cpu::ops::MeanLayer::MeanQuant8 ( )

Definition at line 57 of file MeanLayer.cc.

{
  nnfw::cker::MeanQ8Asymm(getShape(_input), getBuffer<uint8_t>(_input), _input->data_scale(),
                          _input->data_zero_point(), getShape(_output), getBuffer<uint8_t>(_output),
                          _output->data_scale(), _output->data_zero_point(), getReducerAxes(_axes));
}

References _axes, _input, _output, onert::backend::IPortableTensor::data_scale(), onert::backend::IPortableTensor::data_zero_point(), onert::backend::cpu::ops::getReducerAxes(), onert::backend::cpu::ops::getShape(), and nnfw::cker::MeanQ8Asymm().

Referenced by run().

◆ run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 77 of file MeanLayer.cc.

{
  if (_input->data_type() == OperandType::FLOAT32)
  {
    MeanFloat32();
  }
  else if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
  {
    MeanQuant8();
  }
  else
  {
    throw std::runtime_error{"Mean: unsupported data type"};
  }
}