diff options
Diffstat (limited to 'include/armnnTestUtils/MockBackend.hpp')
-rw-r--r-- | include/armnnTestUtils/MockBackend.hpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/include/armnnTestUtils/MockBackend.hpp b/include/armnnTestUtils/MockBackend.hpp index e5378bf74a..05df1b46b1 100644 --- a/include/armnnTestUtils/MockBackend.hpp +++ b/include/armnnTestUtils/MockBackend.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -320,6 +320,11 @@ public: reasonIfUnsupported); } } + case LayerType::ElementwiseBinary: + { + auto elementwiseDesc = *(PolymorphicDowncast<const ElementwiseBinaryDescriptor*>(&descriptor)); + return (elementwiseDesc.m_Operation == BinaryOperation::Add); + } default: return false; } |