onert::backend::cpu::ops::PackLayer Class Reference

#include <PackLayer.h>

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

Public Member Functions
	PackLayer ()
template<typename T >
void	packImpl ()
void	configure (const std::vector< const IPortableTensor * > &inputs, int32_t axis, 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 PackLayer.h.

Constructor & Destructor Documentation

PackLayer()

onert::backend::cpu::ops::PackLayer::PackLayer

(

)

Definition at line 32 of file PackLayer.cc.

                     : _inputs(), _output(nullptr), _axis(0)
{
  // DO NOTHING
}

Member Function Documentation

configure()

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

Definition at line 65 of file PackLayer.cc.

{
  assert(inputs.size() > 0);
  assert(output != nullptr);
 
  _inputs = inputs;
  _axis = axis;
  _output = output;
}

packImpl()

template<typename T >

void onert::backend::cpu::ops::PackLayer::packImpl

(

)

Definition at line 37 of file PackLayer.cc.

{
  uint32_t num_inputs = _inputs.size();
  nnfw::cker::PackParams op_params;
  op_params.axis = _axis;
  op_params.inputs_count = num_inputs;
 
  std::vector<nnfw::cker::Shape *> inputDimsPtr;
  std::vector<nnfw::cker::Shape> inputDims;
  inputDimsPtr.reserve(num_inputs);
  inputDims.reserve(num_inputs);
 
  for (uint32_t i = 0; i < num_inputs; i++)
  {
    inputDims.push_back(getShape(_inputs[i]));
    inputDimsPtr.push_back(&inputDims[i]);
  }
 
  std::vector<const T *> inputPtrs;
 
  for (const auto input : _inputs)
  {
    inputPtrs.emplace_back(getBuffer<T>(input));
  }
 
  nnfw::cker::Pack<T>(op_params, inputPtrs.data(), getShape(_output), getBuffer<T>(_output));
}

References nnfw::cker::PackParams::axis, onert::backend::cpu::ops::getShape(), and nnfw::cker::PackParams::inputs_count.

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 76 of file PackLayer.cc.

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