24 #ifndef ARM_COMPUTE_CPU_POOL2D_H 25 #define ARM_COMPUTE_CPU_POOL2D_H 36 struct PoolingLayerInfo;
75 std::unique_ptr<INEKernel> _pooling_layer_kernel;
76 std::unique_ptr<INEKernel> _border_handler;
77 std::unique_ptr<INEKernel> _asm_glue;
79 bool _is_global_pooling_layer;
Basic interface for functions which have a single async CPU kernel.
Store the tensor's metadata.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
std::vector< MemoryInfo > MemoryRequirements
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
Basic function to simulate a pooling layer with the specified pooling operation.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const PoolingLayerInfo &pool_info, const ITensorInfo *indices=nullptr)
Static function to check if given info will lead to a valid configuration.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Pooling Layer Information struct.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuPool2d)
DataLayout
[DataLayout enum definition]
void configure(ITensorInfo *src, ITensorInfo *dst, const PoolingLayerInfo &pool_info, ITensorInfo *indices=nullptr)
Set the src and dst tensors.