commit a7edbe6d02023c9ad8688c1b12d963361835e139
author Cathal Corbett <cathal.corbett@arm.com> Thu Jan 20 15:53:31 2022 +0000
committer Cathal Corbett <cathal.corbett@arm.com> Fri Jan 21 17:08:03 2022 +0000
tree ec37ea65114b02cfb4ea5d04a3b653e25c93f573
parent 5bcc0723ff66efd46ad46a16b0d8868557f8e37f

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

diff --git a/src/dynamic/sample/SampleDynamicLayerSupport.cpp b/src/dynamic/sample/SampleDynamicLayerSupport.cpp
index cbd221f..3094bc1 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
 {

src/dynamic/sample/SampleDynamicLayerSupport.hpp

diff --git a/src/dynamic/sample/SampleDynamicLayerSupport.hpp b/src/dynamic/sample/SampleDynamicLayerSupport.hpp
index a382073..2f0744a 100644
--- a/src/dynamic/sample/SampleDynamicLayerSupport.hpp
+++ b/src/dynamic/sample/SampleDynamicLayerSupport.hpp
@@ -23,6 +23,13 @@
 
     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