aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/backends/backendsCommon/test/MockBackend.cpp3
-rw-r--r--src/backends/backendsCommon/test/MockBackend.hpp35
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