diff options
Diffstat (limited to 'src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.hpp')
-rw-r--r-- | src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.hpp | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.hpp b/src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.hpp index f4d95d623f..56b0848efa 100644 --- a/src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.hpp +++ b/src/armnn/backends/NeonWorkloads/NeonConvolution2dFloat32Workload.hpp @@ -5,21 +5,25 @@ #pragma once -#include <backends/NeonWorkloadUtils.hpp> #include "NeonConvolution2dBaseWorkload.hpp" +#include <backends/NeonWorkloadUtils.hpp> + +#include "arm_compute/runtime/MemoryManagerOnDemand.h" + +#include <memory> namespace armnn { + class NeonConvolution2dFloat32Workload : public NeonConvolution2dBaseWorkload<DataType::Float32> { public: - NeonConvolution2dFloat32Workload(const Convolution2dQueueDescriptor& descriptor, const WorkloadInfo& info); + NeonConvolution2dFloat32Workload(const Convolution2dQueueDescriptor& descriptor, const WorkloadInfo& info, + std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager); void Execute() const override; void ValidateData() const override; }; -} //namespace armnn - - +} //namespace armnn |