commit 10403ec9f214d4fec2b85866882a9bc707852412
author Tracy Narine <tracy.narine@arm.com> Tue Nov 28 11:55:08 2023 +0000
committer Tracy Narine <tracy.narine@arm.com> Wed Jan 17 20:19:09 2024 +0000
tree cf1bef9015d228f9c84cada5386a085781d38f57
parent 8ded1788e93886c687c86c1136e34cf0ed7acf47

IVGCVSW-7344  Add LeakyRelu Activation support to TOSA Reference Backend

  * Adding a one to many FP32 tosa mapping for Leaky Relu
  * Added a few utilities that are needed
  * Added new tests

Signed-off-by: Tracy Narine <tracy.narine@arm.com>
Change-Id: If1d7c57a523961581777a244416a7346a9310803

src/backends/tosaCommon/TosaMappings.cpp

diff --git a/src/backends/tosaCommon/TosaMappings.cpp b/src/backends/tosaCommon/TosaMappings.cpp
index 6c6bff4..1ebb68b 100644
--- a/src/backends/tosaCommon/TosaMappings.cpp
+++ b/src/backends/tosaCommon/TosaMappings.cpp
@@ -1,5 +1,5 @@
 //
-// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2022-2024 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 
@@ -23,6 +23,18 @@
 {
     switch (type)
     {
+        case LayerType::Activation:
+        {
+            auto activationDesc = PolymorphicDowncast<const ActivationDescriptor*>(&descriptor);
+            if (activationDesc->m_Function == ActivationFunction::LeakyReLu)
+            {
+                return ConvertActivationToTosaOperator(layer, inputs, outputs, activationDesc);
+            }
+            else
+            {
+                return CreateEmptyTosaSerializationBasicBlock();
+            }
+        }
         case LayerType::Addition:
         case LayerType::Multiplication:
         case LayerType::Subtraction: