24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_POOL3D_H
25 #define ARM_COMPUTE_CPU_POOL3D_H
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Basic function to simulate a pooling layer with the specified pooling operation.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const Pooling3dLayerInfo &pool_info)
Static function to check if given info will lead to a valid configuration.
Basic interface for functions which have a single async CPU kernel.
Pooling Layer Information struct.
void configure(const ITensorInfo *src, ITensorInfo *dst, const Pooling3dLayerInfo &pool_info)
Set the src and dst tensors.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuPool3d)
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.