IVGCVSW-2997 Refactor reference LSTM workload

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

1 files changed, 5 insertions, 15 deletions

diff --git a/src/backends/reference/workloads/Activation.cpp b/src/backends/reference/workloads/Activation.cpp
index 760c9a0ccd..2b0c84e226 100644
--- a/src/backends/reference/workloads/Activation.cpp
+++ b/src/backends/reference/workloads/Activation.cpp
@@ -88,26 +88,16 @@ void Activation(Decoder<float>& in,
                 float a,
                 float b)
 {
-    for (size_t i = 0; i<tensorInfo.GetNumElements(); i++)
+    unsigned int numElements = tensorInfo.GetNumElements();
+
+    for (unsigned int i = 0; i < numElements; i++)
     {
         out.Set(Activation(in.Get(), function, a, b));
-
         ++in;
         ++out;
     }
-}
-
-void Activation(const float* in,
-               float* out,
-               const TensorInfo& tensorInfo,
-               ActivationFunction function,
-               float a,
-               float b)
-{
-    for (size_t i = 0; i<tensorInfo.GetNumElements(); i++)
-    {
-        out[i] = Activation(in[i], function, a, b);
-    }
+    in -= numElements;
+    out -= numElements;
 }
 
 } //namespace armnn