diff options
| author | Tracy Narine <tracy.narine@arm.com> | 2023-11-28 11:55:08 +0000 |
|---|---|---|
| committer | Tracy Narine <tracy.narine@arm.com> | 2024-01-17 20:19:09 +0000 |
| commit | 10403ec9f214d4fec2b85866882a9bc707852412 (patch) | |
| tree | cf1bef9015d228f9c84cada5386a085781d38f57 /src/backends/tosaCommon/TosaMappings.cpp | |
| parent | 8ded1788e93886c687c86c1136e34cf0ed7acf47 (diff) | |
| download | armnn-10403ec9f214d4fec2b85866882a9bc707852412.tar.gz | |
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
Diffstat (limited to 'src/backends/tosaCommon/TosaMappings.cpp')
| -rw-r--r-- | src/backends/tosaCommon/TosaMappings.cpp | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/src/backends/tosaCommon/TosaMappings.cpp b/src/backends/tosaCommon/TosaMappings.cpp index 6c6bff4087..1ebb68b3c9 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 @@ TosaSerializationBasicBlock* GetTosaMapping(const Layer* layer, { 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: |