24.02.1
|
Go to the documentation of this file.
39 CpuPool3d::CpuPool3d() : _aux_mem(1)
50 auto k = std::make_unique<kernels::CpuPool3dKernel>();
51 k->configure(
src,
dst, pool_info);
52 _kernel = std::move(k);
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
virtual void schedule_op(ICPPKernel *kernel, const Hints &hints, const Window &window, ITensorPack &tensors)=0
Runs the kernel in the same thread as the caller synchronously.
bool empty() const
Checks if pack is empty.
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
static IScheduler & get()
Access the scheduler singleton.
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.
static constexpr size_t DimY
Alias for dimension 1 also known as Y dimension.
Pooling Layer Information struct.
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.
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.
#define ARM_COMPUTE_LOG_PARAMS(...)
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.