From a7edbe6d02023c9ad8688c1b12d963361835e139 Mon Sep 17 00:00:00 2001
From: Cathal Corbett <cathal.corbett@arm.com>
Date: Thu, 20 Jan 2022 15:53:31 +0000
Subject: IVGCVSW-6735 DynamicBackendTests throwing exception while running
 ArmNN Unittest

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I8f3274c39ba34b031f78c5542da6bf1a495892a9
---
 src/dynamic/sample/SampleDynamicLayerSupport.cpp | 25 ++++++++++++++++++++++++
 src/dynamic/sample/SampleDynamicLayerSupport.hpp |  7 +++++++
 2 files changed, 32 insertions(+)

diff --git a/src/dynamic/sample/SampleDynamicLayerSupport.cpp b/src/dynamic/sample/SampleDynamicLayerSupport.cpp
index cbd221f52b..3094bc1c5b 100644
--- a/src/dynamic/sample/SampleDynamicLayerSupport.cpp
+++ b/src/dynamic/sample/SampleDynamicLayerSupport.cpp
@@ -12,6 +12,31 @@
 namespace sdb // sample dynamic backend
 {
 
+bool SampleDynamicLayerSupport::IsLayerSupported(const armnn::LayerType& type,
+                                                 const std::vector<armnn::TensorInfo>& infos,
+                                                 const armnn::BaseDescriptor& /*descriptor*/,
+                                                 const armnn::Optional<armnn::LstmInputParamsInfo>&
+                                                     /*lstmParamsInfo*/,
+                                                 const armnn::Optional<armnn::QuantizedLstmInputParamsInfo>&
+                                                     /*quantizedLstmParamsInfo*/,
+                                                 armnn::Optional<std::string&> reasonIfUnsupported) const
+{
+    switch (type)
+    {
+        case armnn::LayerType::Input:
+            return IsInputSupported(infos[0], reasonIfUnsupported);
+        case armnn::LayerType::Output:
+            return IsOutputSupported(infos[0], reasonIfUnsupported);
+        case armnn::LayerType::Addition:
+            return IsAdditionSupported(infos[0],
+                                       infos[1],
+                                       infos[2],
+                                       reasonIfUnsupported);
+        default:
+            return false;
+    }
+}
+
 bool SampleDynamicLayerSupport::IsInputSupported(const armnn::TensorInfo& input,
                                                  armnn::Optional<std::string&> reasonIfUnsupported) const
 {
diff --git a/src/dynamic/sample/SampleDynamicLayerSupport.hpp b/src/dynamic/sample/SampleDynamicLayerSupport.hpp
index a38207304f..2f0744aab7 100644
--- a/src/dynamic/sample/SampleDynamicLayerSupport.hpp
+++ b/src/dynamic/sample/SampleDynamicLayerSupport.hpp
@@ -23,6 +23,13 @@ public:
 
     bool IsOutputSupported(const armnn::TensorInfo& output,
                            armnn::Optional<std::string&> reasonIfUnsupported) const override;
+
+    bool IsLayerSupported(const armnn::LayerType& type,
+                          const std::vector<armnn::TensorInfo>& infos,
+                          const armnn::BaseDescriptor& descriptor,
+                          const armnn::Optional<armnn::LstmInputParamsInfo>& lstmParamsInfo,
+                          const armnn::Optional<armnn::QuantizedLstmInputParamsInfo>& quantizedLstmParamsInfo,
+                          armnn::Optional<std::string&> reasonIfUnsupported = armnn::EmptyOptional()) const override;
 };
 
 } // namespace sdb
-- 
cgit v1.2.1