diff options
Diffstat (limited to 'tests/validation/reference')
-rw-r--r-- | tests/validation/reference/ElementwiseOperations.cpp | 7 |
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"); |