24.02.1
|
Basic function to run kernels::CpuMaxUnpoolingLayerKernel. More...
#include <CpuMaxUnpooling.h>
Public Member Functions | |
void | configure (const ITensorInfo *src, const ITensorInfo *indices, ITensorInfo *dst, const PoolingLayerInfo &pool_info) |
Configure operator for a given list of arguments. More... | |
Public Member Functions inherited from INEOperator | |
INEOperator (IRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
INEOperator (const INEOperator &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INEOperator (INEOperator &&)=default | |
Default move constructor. More... | |
INEOperator & | operator= (const INEOperator &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INEOperator & | operator= (INEOperator &&)=default |
Default move assignment operator. More... | |
~INEOperator () | |
Default destructor. 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 *indices, const ITensorInfo *dst, const PoolingLayerInfo &pool_info) |
Static function to check if given info will lead to a valid configuration. More... | |
Basic function to run kernels::CpuMaxUnpoolingLayerKernel.
Definition at line 34 of file CpuMaxUnpooling.h.
void configure | ( | const ITensorInfo * | src, |
const ITensorInfo * | indices, | ||
ITensorInfo * | dst, | ||
const PoolingLayerInfo & | pool_info | ||
) |
Configure operator for a given list of arguments.
[in] | src | Source tensor to permute. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/F32. |
[in] | indices | Tensor containing the offset to store the src elements in the dst tensor. CpuMaxUnpooling 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 src tensor shape. Data type supported: U32. |
[out] | dst | Destination tensor. Data types supported: Same as src |
[in] | pool_info | Contains pooling operation information described in PoolingLayerInfo. |
Definition at line 33 of file CpuMaxUnpooling.cpp.
References ARM_COMPUTE_LOG_PARAMS, arm_compute::test::validation::dst, and arm_compute::test::validation::src.
|
static |
Static function to check if given info will lead to a valid configuration.
Similar to CpuMaxUnpooling::configure()
Definition at line 44 of file CpuMaxUnpooling.cpp.
References arm_compute::test::validation::dst, arm_compute::test::validation::src, and CpuMaxUnpoolingLayerKernel::validate().
Referenced by NEMaxUnpoolingLayer::validate().