aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/reference/ElementwiseOperations.cpp
diff options
context:
space:
mode:
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");