onert::backend::cpu::ops::PoolLayer Class Reference

#include <PoolLayer.h>

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

Public Member Functions
	PoolLayer ()
void	configure (const IPortableTensor *input, const uint32_t paddingLeft, const uint32_t paddingRight, const uint32_t paddingTop, const uint32_t paddingBottom, const uint32_t strideWidth, const uint32_t strideHeight, const uint32_t kernelWidth, const uint32_t kernelHeight, const ir::Activation activation, IPortableTensor *output, const PoolType op_type)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Protected Attributes
const IPortableTensor *	_input
IPortableTensor *	_output

Detailed Description

Definition at line 35 of file PoolLayer.h.

Constructor & Destructor Documentation

PoolLayer()

onert::backend::cpu::ops::PoolLayer::PoolLayer

(

)

Definition at line 64 of file PoolLayer.cc.

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

Member Function Documentation

configure()

void onert::backend::cpu::ops::PoolLayer::configure	(	const IPortableTensor *	input,
		const uint32_t	paddingLeft,
		const uint32_t	paddingRight,
		const uint32_t	paddingTop,
		const uint32_t	paddingBottom,
		const uint32_t	strideWidth,
		const uint32_t	strideHeight,
		const uint32_t	kernelWidth,
		const uint32_t	kernelHeight,
		const ir::Activation	activation,
		IPortableTensor *	output,
		const PoolType	op_type
	)

Definition at line 82 of file PoolLayer.cc.

{
  assert(input != nullptr);
  assert(output != nullptr);
 
  _input = input;
  _output = output;
 
  POOLING_PARAMETERS
 
  switch (_input->data_type())
  {
    case OperandType::FLOAT32:
    {
      float output_activation_min = 0;
      float output_activation_max = 0;
      CalculateActivationRange<float>(activation, &output_activation_min, &output_activation_max);
      op_params.float_activation_min = output_activation_min;
      op_params.float_activation_max = output_activation_max;
 
      _kernel = generateKernelGeneric<float>(op_params, op_type);
      break;
    }
    case OperandType::QUANT_UINT8_ASYMM:
    {
      int32_t output_activation_min = 0;
      int32_t output_activation_max = 0;
      CalculateActivationRangeQuantized(activation, _output, &output_activation_min,
                                        &output_activation_max);
      op_params.quantized_activation_min = output_activation_min;
      op_params.quantized_activation_max = output_activation_max;
      _kernel = generateKernelGeneric<uint8_t>(op_params, op_type);
      break;
    }
    case OperandType::QUANT_INT8_ASYMM:
    {
      int32_t output_activation_min = 0;
      int32_t output_activation_max = 0;
      CalculateActivationRangeQuantized(activation, _output, &output_activation_min,
                                        &output_activation_max);
      op_params.quantized_activation_min = output_activation_min;
      op_params.quantized_activation_max = output_activation_max;
      _kernel = generateKernelGeneric<int8_t>(op_params, op_type);
      break;
    }
    default:
      throw std::runtime_error{"Pool: unsupported data type"};
  }
}

References _input, _output, onert::backend::cpu::ops::CalculateActivationRangeQuantized(), onert::backend::IPortableTensor::data_type(), and POOLING_PARAMETERS.

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 136 of file PoolLayer.cc.

{ _kernel(_input, _output); }

References _input, and _output.

Referenced by onert::backend::train::ops::PoolLayer::forward().

Field Documentation

_input

const IPortableTensor* onert::backend::cpu::ops::PoolLayer::_input

protected

Definition at line 51 of file PoolLayer.h.

Referenced by configure(), onert::backend::train::ops::PoolLayer::forward(), and run().

_output

IPortableTensor* onert::backend::cpu::ops::PoolLayer::_output

protected

Definition at line 52 of file PoolLayer.h.

Referenced by configure(), onert::backend::train::ops::PoolLayer::forward(), and run().

---

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

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