diff options
author | David Monahan <david.monahan@arm.com> | 2019-05-24 10:46:28 +0100 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2019-05-24 10:46:28 +0100 |
commit | 7f2c35a82ec11be50b3478bd15207320bbf3bd57 (patch) | |
tree | 962661c8f80da8f5c011ccebd84df4cc58280bfa /src/backends/backendsCommon/test | |
parent | c1001c6f7c2e622eaff02630e89a35e24778d9b0 (diff) | |
download | armnn-7f2c35a82ec11be50b3478bd15207320bbf3bd57.tar.gz |
IVGCVSW-3033 Adding MockLayerSupport to MockBackend
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I90903af1b8c961a7727da9172a7f02031521a6ad
Diffstat (limited to 'src/backends/backendsCommon/test')
-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 |