aboutsummaryrefslogtreecommitdiff
path: root/include/armnnTestUtils/MockBackend.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnnTestUtils/MockBackend.hpp')
-rw-r--r--include/armnnTestUtils/MockBackend.hpp7
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;
}