IVGCVSW-3033 Adding MockLayerSupport to MockBackend

Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I90903af1b8c961a7727da9172a7f02031521a6ad

2 files changed, 37 insertions, 1 deletions

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