1 files changed, 7 insertions, 1 deletions
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;
         }