21.02
|
Interface for the pooling layer kernel. More...
#include <NEMaxUnpoolingLayerKernel.h>
Public Member Functions | |
const char * | name () const override |
Name of the kernel. More... | |
NEMaxUnpoolingLayerKernel () | |
Default constructor. More... | |
NEMaxUnpoolingLayerKernel (const NEMaxUnpoolingLayerKernel &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEMaxUnpoolingLayerKernel & | operator= (const NEMaxUnpoolingLayerKernel &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEMaxUnpoolingLayerKernel (NEMaxUnpoolingLayerKernel &&)=default | |
Allow instances of this class to be moved. More... | |
NEMaxUnpoolingLayerKernel & | operator= (NEMaxUnpoolingLayerKernel &&)=default |
Allow instances of this class to be moved. More... | |
~NEMaxUnpoolingLayerKernel ()=default | |
Default destructor. More... | |
void | configure (const ITensor *input, const ITensor *indices, ITensor *output, const PoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
void | run (const Window &window, const ThreadInfo &info) override |
Execute the kernel on the passed window. More... | |
Public Member Functions inherited from ICPPKernel | |
virtual | ~ICPPKernel ()=default |
Default destructor. More... | |
virtual void | run_nd (const Window &window, const ThreadInfo &info, const Window &thread_locator) |
legacy compatibility layer for implemantions which do not support thread_locator In these cases we simply narrow the interface down the legacy version More... | |
virtual void | run_op (ITensorPack &tensors, const Window &window, const ThreadInfo &info) |
Execute the kernel on the passed window. More... | |
Public Member Functions inherited from IKernel | |
IKernel () | |
Constructor. More... | |
virtual | ~IKernel ()=default |
Destructor. More... | |
virtual bool | is_parallelisable () const |
Indicates whether or not the kernel is parallelisable. More... | |
virtual BorderSize | border_size () const |
The size of the border for that kernel. More... | |
const Window & | window () const |
The maximum window the kernel can be executed on. More... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, const PoolingLayerInfo &pool_info) |
Static function to check if given info will lead to a valid configuration of NEMaxUnpoolingLayerKernel. More... | |
Interface for the pooling layer kernel.
Definition at line 34 of file NEMaxUnpoolingLayerKernel.h.
Default constructor.
Definition at line 71 of file NEMaxUnpoolingLayerKernel.cpp.
Referenced by NEMaxUnpoolingLayerKernel::name().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Allow instances of this class to be moved.
|
default |
Default destructor.
Referenced by NEMaxUnpoolingLayerKernel::name().
void configure | ( | const ITensor * | input, |
const ITensor * | indices, | ||
ITensor * | output, | ||
const PoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
[in] | input | Source tensor. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/F32. |
[in] | indices | Tensor containing the offset to store the input elements in the output tensor. cpu::kernels::CpuPoolingKernel with indices should precede this function in order to properly reconstruct the output tensor. The tensor shape of this tensor has to be equal to the input tensor shape. Data type supported: U32. |
[out] | output | Destination tensor. Data types supported: Same as input . |
[in] | pool_info | Contains pooling operation information described in PoolingLayerInfo. |
Definition at line 76 of file NEMaxUnpoolingLayerKernel.cpp.
References ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, arm_compute::auto_init_if_empty(), ITensor::buffer(), arm_compute::calculate_max_window(), ICloneable< T >::clone(), arm_compute::misc::shape_calculator::compute_unpool_shape(), ITensorInfo::data_type(), arm_compute::execute_window_loop(), arm_compute::F16, arm_compute::F32, ITensor::info(), arm_compute::test::validation::input, arm_compute::test::validation::output_shape, Iterator::ptr(), arm_compute::QASYMM8, arm_compute::QASYMM8_SIGNED, ITensorInfo::strides_in_bytes(), arm_compute::validate_arguments(), and IKernel::window().
Referenced by NEMaxUnpoolingLayerKernel::name().
|
inlineoverridevirtual |
Name of the kernel.
Implements ICPPKernel.
Definition at line 37 of file NEMaxUnpoolingLayerKernel.h.
References NEMaxUnpoolingLayerKernel::configure(), arm_compute::test::validation::info, arm_compute::test::validation::input, NEMaxUnpoolingLayerKernel::NEMaxUnpoolingLayerKernel(), NEMaxUnpoolingLayerKernel::operator=(), NEMaxUnpoolingLayerKernel::run(), NEMaxUnpoolingLayerKernel::validate(), IKernel::window(), and NEMaxUnpoolingLayerKernel::~NEMaxUnpoolingLayerKernel().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
Referenced by NEMaxUnpoolingLayerKernel::name().
|
default |
Allow instances of this class to be moved.
|
overridevirtual |
Execute the kernel on the passed window.
[in] | window | Region on which to execute the kernel. (Must be a region of the window returned by window()) |
[in] | info | Info about executing thread and CPU. |
Reimplemented from ICPPKernel.
Definition at line 133 of file NEMaxUnpoolingLayerKernel.cpp.
References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW, ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL, ARM_COMPUTE_UNUSED, and IKernel::window().
Referenced by NEMaxUnpoolingLayerKernel::name().
|
static |
Static function to check if given info will lead to a valid configuration of NEMaxUnpoolingLayerKernel.
[in] | input | Source tensor info. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/F32. |
[in] | output | Destination tensor info. Data types supported: Same as input . |
[in] | indices | Tensor info of the indices of the maximal values. Data type supported: U32. |
[in] | pool_info | Contains pooling operation information described in PoolingLayerInfo. |
Definition at line 126 of file NEMaxUnpoolingLayerKernel.cpp.
References ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR, ARM_COMPUTE_RETURN_ON_ERROR, and arm_compute::validate_arguments().
Referenced by NEMaxUnpoolingLayerKernel::name(), and NEMaxUnpoolingLayer::validate().