onert::backend::cpu::ops::DepthToSpaceLayer Class Reference

#include <DepthToSpaceLayer.h>

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

Public Member Functions
	DepthToSpaceLayer ()
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 DepthToSpaceLayer.h.

Constructor & Destructor Documentation

DepthToSpaceLayer()

onert::backend::cpu::ops::DepthToSpaceLayer::DepthToSpaceLayer

(

)

Definition at line 31 of file DepthToSpaceLayer.cc.

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

Member Function Documentation

configure()

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

Definition at line 42 of file DepthToSpaceLayer.cc.

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

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 50 of file DepthToSpaceLayer.cc.

{
  switch (_input->data_type())
  {
    case OperandType::FLOAT32:
      depthToSpace<float>();
      break;
    case OperandType::INT32:
      depthToSpace<int32_t>();
      break;
    case OperandType::INT64:
      depthToSpace<int64_t>();
      break;
    case OperandType::QUANT_UINT8_ASYMM:
      depthToSpace<uint8_t>();
      break;
    case OperandType::QUANT_INT8_ASYMM:
      depthToSpace<int8_t>();
      break;
    default:
      throw std::runtime_error{"DepthToSpace: 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/DepthToSpaceLayer.h
• runtime/onert/backend/cpu/ops/DepthToSpaceLayer.cc