IVGCVSW-2982 Refactor reference Activation workload

Change-Id: Ia3b9a56787cc68822a3c1635de82e03ecc0aae27
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>

1 files changed, 14 insertions, 2 deletions

diff --git a/src/backends/reference/workloads/Activation.hpp b/src/backends/reference/workloads/Activation.hpp
index c8a23114f0..ffe3c5fc5d 100644
--- a/src/backends/reference/workloads/Activation.hpp
+++ b/src/backends/reference/workloads/Activation.hpp
@@ -3,18 +3,30 @@
 // SPDX-License-Identifier: MIT
 //
 
+#include "BaseIterator.hpp"
+
 #include <armnn/Tensor.hpp>
 #include <armnn/Types.hpp>
 
 namespace armnn
 {
+float Activation(float in,
+                 ActivationFunction function,
+                 float a,
+                 float b);
 
-/// Performs the ActivationFunction elementwise on the inputs to give the outputs.
+void Activation(Decoder<float>& in,
+                Encoder<float>& out,
+                const TensorInfo& tensorInfo,
+                ActivationFunction function,
+                float a,
+                float b);
+
+// This is still used by Reference LSTM implementation
 void Activation(const float* in,
                 float* out,
                 const TensorInfo& tensorInfo,
                 ActivationFunction function,
                 float a,
                 float b);
-
 } //namespace armnn