1 files changed, 7 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/ElementwiseFunction.hpp b/src/backends/reference/workloads/ElementwiseFunction.hpp
index 9eb003d5f9..fd1fab0690 100644
--- a/src/backends/reference/workloads/ElementwiseFunction.hpp
+++ b/src/backends/reference/workloads/ElementwiseFunction.hpp
@@ -11,15 +11,18 @@
 namespace armnn
 {
 
-template <typename Functor, typename DecoderOp, typename EncoderOp>
+template <typename Functor>
 struct ElementwiseFunction
 {
+    using OutType = typename Functor::result_type;
+    using InType = typename Functor::first_argument_type;
+
     ElementwiseFunction(const TensorShape& inShape0,
                         const TensorShape& inShape1,
                         const TensorShape& outShape,
-                        DecoderOp& inData0,
-                        DecoderOp& inData1,
-                        EncoderOp& outData);
+                        armnn::Decoder<InType>& inData0,
+                        armnn::Decoder<InType>& inData1,
+                        armnn::Encoder<OutType>& outData);
 };
 
 } //namespace armnn