onert::backend::train::ops::PoolLayer Class Reference

#include <PoolLayer.h>

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

Public Member Functions
	PoolLayer ()
void	configureBackward (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, const PoolType op_type, IPortableTensor *output, IPortableTensor *back_prop_input, const IPortableTensor *back_prop_output)
void	forward (bool training) override
void	backward () override
Public Member Functions inherited from onert::exec::train::ITrainableFunction
virtual	~ITrainableFunction ()=default
virtual std::optional< backend::train::LayerScopeTensors >	registerLayerScopeTensors ()
Public Member Functions inherited from onert::backend::cpu::ops::PoolLayer
	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 ()

Additional Inherited Members
Protected Attributes inherited from onert::backend::cpu::ops::PoolLayer
const IPortableTensor *	_input
IPortableTensor *	_output

Detailed Description

Definition at line 51 of file PoolLayer.h.

Constructor & Destructor Documentation

PoolLayer()

onert::backend::train::ops::PoolLayer::PoolLayer

(

)

Definition at line 188 of file PoolLayer.cc.

  : cpu::ops::PoolLayer(), _back_prop_input(nullptr), _back_prop_output(nullptr), _kernel(nullptr)
{
  // DO NOTHING
}

Member Function Documentation

backward()

void onert::backend::train::ops::PoolLayer::backward ( )

overridevirtual

Implements onert::exec::train::ITrainableFunction.

Definition at line 240 of file PoolLayer.cc.

{ _kernel->backward(_back_prop_output, _back_prop_input); }

configureBackward()

void onert::backend::train::ops::PoolLayer::configureBackward	(	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,
		const PoolType	op_type,
		IPortableTensor *	output,
		IPortableTensor *	back_prop_input,
		const IPortableTensor *	back_prop_output
	)

Definition at line 194 of file PoolLayer.cc.

{
  _back_prop_output = back_prop_output;
  _back_prop_input = back_prop_input;
 
  if (output->data_type() != OperandType::FLOAT32)
  {
    throw std::runtime_error("PoolLayer : Unsupported data type for training");
  }
 
  // ready training kernel
  switch (op_type)
  {
    case PoolType::kMax:
      _kernel = std::make_unique<MaxPool2D>(paddingLeft, paddingRight, paddingTop, paddingBottom,
                                            strideWidth, strideHeight, kernelWidth, kernelHeight,
                                            activation, output);
      break;
    case PoolType::kAvg:
      _kernel = std::make_unique<AveragePool2D>(paddingLeft, paddingRight, paddingTop,
                                                paddingBottom, strideWidth, strideHeight,
                                                kernelWidth, kernelHeight, activation, output);
      break;
    default:
      throw std::runtime_error("PoolLayer: Unsupported pool type");
  }
}

References onert::backend::train::ops::kAvg, and onert::backend::train::ops::kMax.

forward()

void onert::backend::train::ops::PoolLayer::forward ( bool  training )

overridevirtual

Implements onert::exec::train::ITrainableFunction.

Definition at line 228 of file PoolLayer.cc.

{
  if (training)
  {
    _kernel->forward(_input, _output);
  }
  else
  {
    cpu::ops::PoolLayer::run();
  }
}

References onert::backend::cpu::ops::PoolLayer::_input, onert::backend::cpu::ops::PoolLayer::_output, and onert::backend::cpu::ops::PoolLayer::run().

---

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

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