diff options
Diffstat (limited to 'src/backends/backendsCommon')
-rw-r--r-- | src/backends/backendsCommon/test/MockBackend.cpp | 3 | ||||
-rw-r--r-- | src/backends/backendsCommon/test/MockBackend.hpp | 35 |
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 |