diff options
Diffstat (limited to 'src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.hpp')
-rw-r--r-- | src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.hpp | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.hpp b/src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.hpp index 98d075a5ea..d28d50d819 100644 --- a/src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.hpp +++ b/src/armnn/backends/NeonWorkloads/NeonConvolution2dBaseWorkload.hpp @@ -12,16 +12,27 @@ #include "backends/ArmComputeTensorUtils.hpp" #include "backends/NeonLayerSupport.hpp" +#include "arm_compute/runtime/MemoryManagerOnDemand.h" + +#include <memory> + namespace armnn { +arm_compute::Status NeonConvolution2dWorkloadValidate(const TensorInfo& input, + const TensorInfo& output, + const Convolution2dDescriptor& descriptor, + const TensorInfo& weights, + const TensorInfo& biases); + template<armnn::DataType dataType> class NeonConvolution2dBaseWorkload : public TypedWorkload<Convolution2dQueueDescriptor, dataType> { public: using TypedWorkload<Convolution2dQueueDescriptor, dataType>::m_Data; - NeonConvolution2dBaseWorkload(const Convolution2dQueueDescriptor& descriptor, const WorkloadInfo& info); + NeonConvolution2dBaseWorkload(const Convolution2dQueueDescriptor& descriptor, const WorkloadInfo& info, + std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager); virtual void ValidateData() const {}; @@ -30,4 +41,5 @@ protected: arm_compute::Tensor m_KernelTensor; arm_compute::Tensor m_BiasTensor; }; -} //namespace armnn
\ No newline at end of file + +} //namespace armnn |