IVGCVSW-3033 Adding MockLayerSupport to MockBackend
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I90903af1b8c961a7727da9172a7f02031521a6ad
diff --git a/src/backends/backendsCommon/test/MockBackend.hpp b/src/backends/backendsCommon/test/MockBackend.hpp
index d0a3de8..88c4785 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 @@
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