2 files changed, 5 insertions, 1 deletions
diff --git a/tests/datasets/ActivationFunctionsDataset.h b/tests/datasets/ActivationFunctionsDataset.h
index 147c5ae51b..c5dc28f0c3 100644
--- a/tests/datasets/ActivationFunctionsDataset.h
+++ b/tests/datasets/ActivationFunctionsDataset.h
@@ -51,7 +51,8 @@ public:
                             ActivationLayerInfo::ActivationFunction::SOFT_RELU,
                             ActivationLayerInfo::ActivationFunction::SQRT,
                             ActivationLayerInfo::ActivationFunction::SQUARE,
-                            ActivationLayerInfo::ActivationFunction::TANH
+                            ActivationLayerInfo::ActivationFunction::TANH,
+                            ActivationLayerInfo::ActivationFunction::IDENTITY
     })
     {
     }
diff --git a/tests/validation/reference/ActivationLayer.h b/tests/validation/reference/ActivationLayer.h
index 77b353056e..cc43bc10a8 100644
--- a/tests/validation/reference/ActivationLayer.h
+++ b/tests/validation/reference/ActivationLayer.h
@@ -75,6 +75,9 @@ inline T activate_float(T x, T a, T b, ActivationLayerInfo::ActivationFunction a
         case ActivationLayerInfo::ActivationFunction::TANH:
             ret = a * std::tanh(b * x);
             break;
+        case ActivationLayerInfo::ActivationFunction::IDENTITY:
+            ret = x;
+            break;
         default:
             ARM_COMPUTE_ERROR("Unsupported activation function");
             break;