10 #include <arm_compute/core/Error.h> 11 #include <arm_compute/runtime/IFunction.h> 19 const TensorInfo& output,
32 std::unique_ptr<arm_compute::IFunction> m_PoolingLayer;
NeonPooling2dWorkload(const Pooling2dQueueDescriptor &descriptor, const WorkloadInfo &info)
Copyright (c) 2021 ARM Limited and Contributors.
const armnnSerializer::Pooling2dDescriptor * Pooling2dDescriptor
Contains information about TensorInfos of a layer.
void Execute() const override
arm_compute::Status NeonPooling2dWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const Pooling2dDescriptor &descriptor)