1 files changed, 3 insertions, 17 deletions
diff --git a/src/backends/reference/workloads/RefConstantWorkload.hpp b/src/backends/reference/workloads/RefConstantWorkload.hpp
index 75d7ecce26..ada488a7b2 100644
--- a/src/backends/reference/workloads/RefConstantWorkload.hpp
+++ b/src/backends/reference/workloads/RefConstantWorkload.hpp
@@ -14,27 +14,13 @@ namespace armnn
 {
 
 // Base class template providing an implementation of the Constant layer common to all data types.
-template <armnn::DataType DataType>
-class RefConstantWorkload : public TypedWorkload<ConstantQueueDescriptor, DataType>
+class RefConstantWorkload : public BaseWorkload<ConstantQueueDescriptor>
 {
 public:
-    RefConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info)
-        : TypedWorkload<ConstantQueueDescriptor, DataType>(descriptor, info)
-        , m_RanOnce(false)
-    {
-    }
-
-    using TypedWorkload<ConstantQueueDescriptor, DataType>::m_Data;
-    using TypedWorkload<ConstantQueueDescriptor, DataType>::TypedWorkload;
+    RefConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info);
 
+    void PostAllocationConfigure() override;
     virtual void Execute() const override;
-
-private:
-    mutable bool m_RanOnce;
 };
 
-using RefConstantFloat32Workload = RefConstantWorkload<DataType::Float32>;
-using RefConstantUint8Workload = RefConstantWorkload<DataType::QuantisedAsymm8>;
-using RefConstantInt32Workload = RefConstantWorkload<DataType::Signed32>;
-
 } //namespace armnn