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/MockBackend.hpp | |
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/MockBackend.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/MockBackend.hpp | 35 |
1 files changed, 35 insertions, 0 deletions
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 |