aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/mockBackend/MockImportLayerSupport.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/test/mockBackend/MockImportLayerSupport.hpp')
-rw-r--r--src/backends/backendsCommon/test/mockBackend/MockImportLayerSupport.hpp9
1 files changed, 7 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/test/mockBackend/MockImportLayerSupport.hpp b/src/backends/backendsCommon/test/mockBackend/MockImportLayerSupport.hpp
index 380ce4a3f5..da4b7ab7d0 100644
--- a/src/backends/backendsCommon/test/mockBackend/MockImportLayerSupport.hpp
+++ b/src/backends/backendsCommon/test/mockBackend/MockImportLayerSupport.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
@@ -16,7 +16,7 @@ class MockImportLayerSupport : public LayerSupportBase
public:
bool IsLayerSupported(const LayerType& type,
const std::vector<TensorInfo>& infos,
- const BaseDescriptor& /*descriptor*/,
+ const BaseDescriptor& descriptor,
const Optional<LstmInputParamsInfo>& /*lstmParamsInfo*/,
const Optional<QuantizedLstmInputParamsInfo>& /*quantizedLstmParamsInfo*/,
Optional<std::string&> reasonIfUnsupported) const override
@@ -25,6 +25,11 @@ public:
{
case LayerType::Addition:
return IsAdditionSupported(infos[0], infos[1], infos[2], reasonIfUnsupported);
+ case LayerType::ElementwiseBinary:
+ {
+ auto elementwiseDesc = *(PolymorphicDowncast<const ElementwiseBinaryDescriptor*>(&descriptor));
+ return (elementwiseDesc.m_Operation == BinaryOperation::Add);
+ }
case LayerType::Input:
return IsInputSupported(infos[0], reasonIfUnsupported);
case LayerType::Output: