24.02.1
|
Function to perform MaxUnpooling. More...
#include <CLMaxUnpoolingLayer.h>
Public Member Functions | |
CLMaxUnpoolingLayer () | |
Constructor. More... | |
CLMaxUnpoolingLayer (const CLMaxUnpoolingLayer &)=delete | |
Prevent instances of this class from being copied. More... | |
CLMaxUnpoolingLayer & | operator= (const CLMaxUnpoolingLayer &)=delete |
Prevent instances of this class from being copied. More... | |
~CLMaxUnpoolingLayer () | |
Default destructor. More... | |
void | configure (ICLTensor *input, ICLTensor *indices, ICLTensor *output, const PoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
void | configure (const CLCompileContext &compile_context, ICLTensor *input, ICLTensor *indices, ICLTensor *output, const PoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
void | run () override |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
virtual void | prepare () |
Prepare the function for executing. 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 CLMaxUnpoolingLayer. More... | |
Function to perform MaxUnpooling.
This function calls the following OpenCL kernels:
Definition at line 46 of file CLMaxUnpoolingLayer.h.
Constructor.
Definition at line 36 of file CLMaxUnpoolingLayer.cpp.
|
delete |
Prevent instances of this class from being copied.
|
default |
Default destructor.
void configure | ( | const CLCompileContext & | compile_context, |
ICLTensor * | input, | ||
ICLTensor * | indices, | ||
ICLTensor * | output, | ||
const PoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
[in] | compile_context | The compile context to be used. |
[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. CLPoolingLayer 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 51 of file CLMaxUnpoolingLayer.cpp.
References ARM_COMPUTE_LOG_PARAMS, CLFill::configure(), and arm_compute::test::validation::input.
void configure | ( | ICLTensor * | input, |
ICLTensor * | indices, | ||
ICLTensor * | output, | ||
const PoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
Valid data layouts:
Valid data type configurations:
src | dst |
---|---|
QASYMM8 | QASYMM8 |
QASYMM8_SIGNED | QASYMM8_SIGNED |
F16 | F16 |
F32 | F32 |
[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. CLPoolingLayer 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 43 of file CLMaxUnpoolingLayer.cpp.
References CLKernelLibrary::get(), and arm_compute::test::validation::input.
|
delete |
Prevent instances of this class from being copied.
|
overridevirtual |
Run the kernels contained in the function.
For CPU kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 72 of file CLMaxUnpoolingLayer.cpp.
References CLScheduler::enqueue(), CLScheduler::get(), and CLFill::run().
|
static |
Static function to check if given info will lead to a valid configuration of CLMaxUnpoolingLayer.
[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 | TensorInfo associated to the tensor containing the offset to store the input elements in the output tensor. CLPoolingLayer 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. |
[in] | pool_info | Contains pooling operation information described in PoolingLayerInfo. |
Definition at line 64 of file CLMaxUnpoolingLayer.cpp.
References arm_compute::test::validation::input, and CLMaxUnpoolingLayerKernel::validate().