24.02.1
|
Basic function to simulate a pooling layer with the specified pooling operation. More...
#include <ClPool2d.h>
Public Member Functions | |
ClPool2d ()=default | |
Constructor. More... | |
void | configure (const ClCompileContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const PoolingLayerInfo &info, ITensorInfo *indices=nullptr) |
Configure operator for a given list of arguments. More... | |
Public Member Functions inherited from ICLOperator | |
ICLOperator (IRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
ICLOperator (const ICLOperator &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
ICLOperator (ICLOperator &&)=default | |
Default move constructor. More... | |
ICLOperator & | operator= (const ICLOperator &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
ICLOperator & | operator= (ICLOperator &&)=default |
Default move assignment operator. More... | |
void | run (ITensorPack &tensors) override |
Run the kernels contained in the function. More... | |
void | prepare (ITensorPack &constants) override |
Prepare the function for executing. More... | |
MemoryRequirements | workspace () const override |
Return the memory requirements required by the workspace. More... | |
Public Member Functions inherited from IOperator | |
virtual | ~IOperator ()=default |
Destructor. More... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *src, const ITensorInfo *dst, const PoolingLayerInfo &info, const ITensorInfo *indices=nullptr) |
Static function to check if given info will lead to a valid configuration. More... | |
Basic function to simulate a pooling layer with the specified pooling operation.
This function calls the following OpenCL kernels:
Definition at line 40 of file ClPool2d.h.
|
default |
Constructor.
void configure | ( | const ClCompileContext & | compile_context, |
ITensorInfo * | src, | ||
ITensorInfo * | dst, | ||
const PoolingLayerInfo & | info, | ||
ITensorInfo * | indices = nullptr |
||
) |
Configure operator for a given list of arguments.
[in] | compile_context | The compile context to be used. |
[in] | src | Source tensor info. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/F32. |
[out] | dst | Destination tensor info. Data type supported: same as src |
[in] | info | Pooling layer parameters. |
[out] | indices | (optional) The indices info of the maximal values. Data type supported: U32. |
Definition at line 36 of file ClPool2d.cpp.
References ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_LOG_PARAMS, arm_compute::test::validation::dst, CLScheduler::get(), arm_compute::test::validation::info, arm_compute::test::validation::src, and CLScheduler::tune_kernel_static().
|
static |
Static function to check if given info will lead to a valid configuration.
Similar to ClPool2d::configure()
Definition at line 55 of file ClPool2d.cpp.
References arm_compute::test::validation::dst, arm_compute::test::validation::info, arm_compute::test::validation::src, and ClPool2dKernel::validate().
Referenced by CLPoolingLayer::validate().