diff options
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: |