commit 3b3c381963a1bfe12e083928a3abb5a9852b199b
author David Monahan <david.monahan@arm.com> Tue Feb 25 09:03:29 2020 +0000
committer David Monahan <david.monahan@arm.com> Fri Feb 28 09:47:17 2020 +0000
tree 7f6b2d7416240490172e01070953d949d32b4407
parent dd3f71b64072c44cec65a7a883d0c3a29659645c

IVGCVSW-4439: Adding Elu support to Activation

 * Added CpuRef implementation
 * Added Unit Tests
 * Added Quantizer Test
 * Enabled Tests for Neon and CL backends on fp32 only
 * Added to Serializer

Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: Ic23e1797dbc9352b40678c389d7fe2b836b582ea

src/backends/reference/workloads/Activation.cpp

diff --git a/src/backends/reference/workloads/Activation.cpp b/src/backends/reference/workloads/Activation.cpp
index 814a0dd..82dd919 100644
--- a/src/backends/reference/workloads/Activation.cpp
+++ b/src/backends/reference/workloads/Activation.cpp
@@ -69,6 +69,11 @@
             output = a * tanhf(b * in);
             break;
         }
+        case ActivationFunction::Elu:
+        {
+            output = (in >= 0) ? in : a * (expf(in) - 1);
+            break;
+        }
         default:
         {
             throw InvalidArgumentException("Unsupported activation function");