aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp')
-rw-r--r--src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp16
1 files changed, 3 insertions, 13 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp b/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp
index ee09ff3e58..a4277d405f 100644
--- a/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp
+++ b/src/armnn/backends/ClWorkloads/ClDepthwiseConvolutionUint8Workload.hpp
@@ -5,29 +5,19 @@
#pragma once
+#include "ClDepthwiseConvolutionBaseWorkload.hpp"
+
#include "backends/ClWorkloadUtils.hpp"
namespace armnn
{
-class ClDepthwiseConvolutionUint8Workload : public Uint8Workload<DepthwiseConvolution2dQueueDescriptor>
+class ClDepthwiseConvolutionUint8Workload : public ClDepthwiseConvolutionBaseWorkload<DataType::QuantisedAsymm8>
{
public:
ClDepthwiseConvolutionUint8Workload(const DepthwiseConvolution2dQueueDescriptor& descriptor,
const WorkloadInfo& info);
void Execute() const override;
-
-private:
- typedef uint8_t KernelDataType;
- typedef int32_t BiasDataType;
-
- mutable std::unique_ptr<arm_compute::IFunction> m_pDepthwiseConvolutionLayer;
-
- arm_compute::CLTensor m_KernelTensor;
- arm_compute::CLTensor m_BiasTensor;
-
- template <typename WorkloadType>
- friend void InitClDepthwiseConvolutionWorkload(WorkloadType& workload);
};
} //namespace armnn