From 1e6e1b899c1a88d3466cdc6fd097ccf32ff767e3 Mon Sep 17 00:00:00 2001
From: giuros01 <giuseppe.rossini@arm.com>
Date: Tue, 14 May 2019 16:12:53 +0100
Subject: COMPMID-2322: PRELU support in CLActivationLayer

Change-Id: I3aa8d4964a1861c2b8acef59dc863df7c48f3316
Signed-off-by: giuros01 <giuseppe.rossini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1146
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
---
 tests/validation/reference/ElementwiseOperations.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'tests/validation/reference/ElementwiseOperations.cpp')

diff --git a/tests/validation/reference/ElementwiseOperations.cpp b/tests/validation/reference/ElementwiseOperations.cpp
index 82f42a0c21..44eb417969 100644
--- a/tests/validation/reference/ElementwiseOperations.cpp
+++ b/tests/validation/reference/ElementwiseOperations.cpp
@@ -81,6 +81,13 @@ T arithm_op(ArithmeticOperation op, T src1, T src2, ConvertPolicy convert_policy
             val = std::pow(static_cast<intermediate_type>(src1), static_cast<intermediate_type>(src2));
             break;
         }
+        case ArithmeticOperation::PRELU:
+        {
+            const T x     = static_cast<intermediate_type>(src1);
+            const T alpha = static_cast<intermediate_type>(src2);
+            val           = (x > 0 ? x : alpha * x);
+            break;
+        }
         default:
         {
             ARM_COMPUTE_ERROR("Not handled");
-- 
cgit v1.2.1