From 7f2c35a82ec11be50b3478bd15207320bbf3bd57 Mon Sep 17 00:00:00 2001
From: David Monahan <david.monahan@arm.com>
Date: Fri, 24 May 2019 10:46:28 +0100
Subject: IVGCVSW-3033 Adding MockLayerSupport to MockBackend

Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I90903af1b8c961a7727da9172a7f02031521a6ad
---
 src/backends/backendsCommon/test/MockBackend.cpp |  3 +-
 src/backends/backendsCommon/test/MockBackend.hpp | 35 ++++++++++++++++++++++++
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/src/backends/backendsCommon/test/MockBackend.cpp b/src/backends/backendsCommon/test/MockBackend.cpp
index 4eb96b5336..043b5125c7 100644
--- a/src/backends/backendsCommon/test/MockBackend.cpp
+++ b/src/backends/backendsCommon/test/MockBackend.cpp
@@ -109,7 +109,8 @@ IBackendInternal::Optimizations MockBackend::GetOptimizations() const
 
 IBackendInternal::ILayerSupportSharedPtr MockBackend::GetLayerSupport() const
 {
-    return ILayerSupportSharedPtr{};
+    static ILayerSupportSharedPtr layerSupport{new MockLayerSupport};
+    return layerSupport;
 }
 
 OptimizationViews MockBackend::OptimizeSubgraphView(const SubgraphView& subgraph) const
diff --git a/src/backends/backendsCommon/test/MockBackend.hpp b/src/backends/backendsCommon/test/MockBackend.hpp
index d0a3de8682..88c47850cf 100644
--- a/src/backends/backendsCommon/test/MockBackend.hpp
+++ b/src/backends/backendsCommon/test/MockBackend.hpp
@@ -7,6 +7,8 @@
 
 #include <backendsCommon/IBackendInternal.hpp>
 #include <backendsCommon/OptimizationViews.hpp>
+#include <LayerSupportCommon.hpp>
+#include <backendsCommon/LayerSupportBase.hpp>
 
 namespace armnn
 {
@@ -33,4 +35,37 @@ public:
     OptimizationViews OptimizeSubgraphView(const SubgraphView& subgraph) const override;
 };
 
+class MockLayerSupport : public LayerSupportBase {
+public:
+    bool IsInputSupported(const TensorInfo& input,
+                          Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override
+    {
+        return true;
+    }
+
+    bool IsOutputSupported(const TensorInfo& input,
+                          Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override
+    {
+        return true;
+    }
+
+    bool IsAdditionSupported(const TensorInfo& input0,
+                             const TensorInfo& input1,
+                             const TensorInfo& output,
+                             Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override
+    {
+        return true;
+    }
+
+    bool IsConvolution2dSupported(const TensorInfo& input,
+                                  const TensorInfo& output,
+                                  const Convolution2dDescriptor& descriptor,
+                                  const TensorInfo& weights,
+                                  const Optional<TensorInfo>& biases,
+                                  Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override
+    {
+        return true;
+    }
+};
+
 } // namespace armnn
-- 
cgit v1.2.1