onert::backend::cpu::ops::SpaceToDepthLayer Class Reference

#include <SpaceToDepthLayer.h>

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

Public Member Functions
	SpaceToDepthLayer ()
void	configure (const IPortableTensor *input, const int32_t block_size, 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 SpaceToDepthLayer.h.

Constructor & Destructor Documentation

SpaceToDepthLayer()

onert::backend::cpu::ops::SpaceToDepthLayer::SpaceToDepthLayer

(

)

Definition at line 31 of file SpaceToDepthLayer.cc.

                                     : _input(nullptr), _block_size(0), _output(nullptr)
{
  // DO NOTHING
}

Member Function Documentation

configure()

void onert::backend::cpu::ops::SpaceToDepthLayer::configure	(	const IPortableTensor *	input,
		const int32_t	block_size,
		IPortableTensor *	output
	)

Definition at line 46 of file SpaceToDepthLayer.cc.

{
  _input = input;
  _block_size = block_size;
  _output = output;
}

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 54 of file SpaceToDepthLayer.cc.

{
  if (_input->data_type() == OperandType::FLOAT32)
  {
    spaceToDepth<float>();
  }
  else if (_input->data_type() == OperandType::QUANT_UINT8_ASYMM)
  {
    spaceToDepth<uint8_t>();
  }
  else
  {
    throw std::runtime_error{"SpaceToDepth: 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/SpaceToDepthLayer.h
• runtime/onert/backend/cpu/ops/SpaceToDepthLayer.cc