diff options
Diffstat (limited to 'src/armnn/optimizations/MovePermuteUp.hpp')
-rw-r--r-- | src/armnn/optimizations/MovePermuteUp.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
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; } |