aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/reference/ElementwiseOperations.cpp
diff options
context:
space:
mode:
authorgiuros01 <giuseppe.rossini@arm.com>2019-05-14 16:12:53 +0100
committerGiuseppe Rossini <giuseppe.rossini@arm.com>2019-05-24 09:20:27 +0000
commit1e6e1b899c1a88d3466cdc6fd097ccf32ff767e3 (patch)
tree124d00f9b2bd26ddbb5edf8f3bc7102e7494d404 /tests/validation/reference/ElementwiseOperations.cpp
parent048b0f300ee729cac1b3019311589c654771fb8f (diff)
downloadComputeLibrary-1e6e1b899c1a88d3466cdc6fd097ccf32ff767e3.tar.gz
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>
Diffstat (limited to 'tests/validation/reference/ElementwiseOperations.cpp')
-rw-r--r--tests/validation/reference/ElementwiseOperations.cpp7
1 files changed, 7 insertions, 0 deletions
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");