diff options
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 |