onert::backend::cpu::ops::AddNLayer Class Reference

#include <AddNLayer.h>

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

Public Member Functions
	AddNLayer ()
void	configure (std::vector< const IPortableTensor * > &&inputs, IPortableTensor *output)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 33 of file AddNLayer.h.

Constructor & Destructor Documentation

AddNLayer()

onert::backend::cpu::ops::AddNLayer::AddNLayer ( )

inline

Definition at line 36 of file AddNLayer.h.

: _inputs(), _output(nullptr) {}

Member Function Documentation

configure()

void onert::backend::cpu::ops::AddNLayer::configure	(	std::vector< const IPortableTensor * > &&	inputs,
		IPortableTensor *	output
	)

Definition at line 33 of file AddNLayer.cc.

{
  _inputs = std::move(inputs);
  _output = output;
}

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 39 of file AddNLayer.cc.

{
  size_t input_size = _inputs.size();
  if (_output->data_type() == ir::DataType::INT32)
  {
    std::vector<const int32_t *> input_buffers(input_size);
    for (size_t i = 0; i < input_size; i++)
    {
      input_buffers[i] = getBuffer<int32_t>(_inputs[i]);
    }
    AddN(getShape(_inputs[0]), input_size, input_buffers.data(), getBuffer<int32_t>(_output));
  }
  else if (_output->data_type() == ir::DataType::FLOAT32)
  {
    std::vector<const float *> input_buffers(input_size);
    for (size_t i = 0; i < input_size; i++)
    {
      input_buffers[i] = getBuffer<float>(_inputs[i]);
    }
    AddN(getShape(_inputs[0]), input_size, input_buffers.data(), getBuffer<float>(_output));
  }
  else
  {
    throw std::runtime_error("AddN: unsupported data type");
  }
}

References onert::backend::IPortableTensor::data_type(), and onert::backend::cpu::ops::getShape().

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

---

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

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