24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_POOL2D_H
25 #define ARM_COMPUTE_CPU_POOL2D_H
37 struct PoolingLayerInfo;
80 std::unique_ptr<INEKernel> _pooling_layer_kernel;
81 std::unique_ptr<INEKernel> _asm_glue;
83 bool _is_global_pooling_layer;
84 bool _use_kernel_indices;
void configure(ITensorInfo *src, ITensorInfo *dst, const PoolingLayerInfo &pool_info, ITensorInfo *indices=nullptr)
Set the src and dst tensors.
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
DataLayout
[DataLayout enum definition]
Basic function to simulate a pooling layer with the specified pooling operation.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Pooling Layer Information struct.
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.
Basic interface for functions which have a single async CPU kernel.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuPool2d)
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.