diff options
Diffstat (limited to 'src/armnn/optimizations')
-rw-r--r-- | src/armnn/optimizations/AddBroadcastReshapeLayer.hpp | 4 | ||||
-rw-r--r-- | src/armnn/optimizations/MovePermuteUp.hpp | 8 | ||||
-rw-r--r-- | src/armnn/optimizations/MoveTransposeUp.hpp | 8 |
3 files changed, 16 insertions, 4 deletions
diff --git a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp index b9e8584a5a..dbde72b917 100644 --- a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp +++ b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2020-2021,2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -18,7 +18,7 @@ namespace optimizations static const std::set<armnn::LayerType> broadcastOps{ LayerType::Addition, LayerType::Division, LayerType::Maximum, LayerType::Minimum, LayerType::Multiplication, LayerType::Prelu, - LayerType::Subtraction }; + LayerType::Subtraction, LayerType::ElementwiseBinary }; class AddBroadcastReshapeLayerImpl { diff --git a/src/armnn/optimizations/MovePermuteUp.hpp b/src/armnn/optimizations/MovePermuteUp.hpp index ae8a28cba4..19078b3bc6 100644 --- a/src/armnn/optimizations/MovePermuteUp.hpp +++ b/src/armnn/optimizations/MovePermuteUp.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017-2018,2020,2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -68,6 +68,12 @@ private: case LayerType::MemCopy: case LayerType::Multiplication: return true; + case LayerType::ElementwiseBinary: + { + auto descriptor = PolymorphicDowncast<const ElementwiseBinaryDescriptor*>(&base.GetParameters()); + return (descriptor->m_Operation == BinaryOperation::Add || + descriptor->m_Operation == BinaryOperation::Mul); + } default: return false; } diff --git a/src/armnn/optimizations/MoveTransposeUp.hpp b/src/armnn/optimizations/MoveTransposeUp.hpp index 999a4ebe42..40f6b9cdac 100644 --- a/src/armnn/optimizations/MoveTransposeUp.hpp +++ b/src/armnn/optimizations/MoveTransposeUp.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2020 Arm Ltd. All rights reserved. +// Copyright © 2020,2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -68,6 +68,12 @@ private: case LayerType::MemCopy: case LayerType::Multiplication: return true; + case LayerType::ElementwiseBinary: + { + auto descriptor = PolymorphicDowncast<const ElementwiseBinaryDescriptor*>(&base.GetParameters()); + return (descriptor->m_Operation == BinaryOperation::Add || + descriptor->m_Operation == BinaryOperation::Mul); + } default: return false; } |