onert::backend::cpu::ops::UnpackLayer Class Reference

#include <UnpackLayer.h>

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

Public Member Functions
	UnpackLayer ()
void	configure (const IPortableTensor *input, uint32_t axis, int32_t num_output, std::vector< 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 UnpackLayer.h.

Constructor & Destructor Documentation

UnpackLayer()

onert::backend::cpu::ops::UnpackLayer::UnpackLayer

(

)

Definition at line 32 of file UnpackLayer.cc.

                         : _input(nullptr), _outputs(), _axis(0), _num_output(0)
{
  // DO NOTHING
}

Member Function Documentation

configure()

void onert::backend::cpu::ops::UnpackLayer::configure	(	const IPortableTensor *	input,
		uint32_t	axis,
		int32_t	num_output,
		std::vector< IPortableTensor * > &	output
	)

Definition at line 65 of file UnpackLayer.cc.

{
  assert(input != nullptr);
  assert(outputs.size() > 0);
  assert(outputs.size() == (size_t)num);
 
  _input = input;
  _axis = axis;
  _num_output = num;
  _outputs = outputs;
}

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 78 of file UnpackLayer.cc.

{
  if (_input->data_type() == OperandType::FLOAT32)
    unpackImpl<float>();
  else if (_input->data_type() == OperandType::INT32)
    unpackImpl<int32_t>();
  else
    throw std::runtime_error{"Unpack: 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/UnpackLayer.h
• runtime/onert/backend/cpu/ops/UnpackLayer.cc