24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_POOL2D_ASSEMBLY_WRAPPER_KERNEL_H
25 #define ARM_COMPUTE_CPU_POOL2D_ASSEMBLY_WRAPPER_KERNEL_H
30 #include "src/core/NEON/kernels/assembly/pooling.hpp"
34 #include "pool_common.hpp"
58 const char *
name()
const override
60 return "CpuPool2dAssemblyWrapperKernel";
104 template <
typename Typesrc,
typename Typedst>
114 template <
typename Typesrc,
typename Typedst>
120 std::unique_ptr<arm_conv::pooling::IPoolingCommon> _kernel_asm{
nullptr};
129 size_t get_mws(
const CPUInfo &platform,
size_t thread_count)
const override;
SimpleTensor< float > src
bool is_configured() const
Was the asm kernel successfully configured?
size_t get_working_size(unsigned int num_threads) const
Get size of the workspace needed by the assembly kernel.
CpuPool2dAssemblyWrapperKernel()=default
Constructor.
This class is a wrapper for the assembly kernels.
Pooling Layer Information struct.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const PoolingLayerInfo &info)
Static function to check if given info will lead to a valid configuration.
void configure(const ITensorInfo *src, ITensorInfo *dst, const PoolingLayerInfo &info, const CPUInfo &cpu_info)
Initialise the kernel's src and dst.
const char * name() const override
Name of the kernel.
const Window & window() const
The maximum window the kernel can be executed on.
Information about executing thread and CPU.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuPool2dAssemblyWrapperKernel)
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)