diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-01-20 15:53:31 +0000 |
---|---|---|
committer | Cathal Corbett <cathal.corbett@arm.com> | 2022-01-21 17:08:03 +0000 |
commit | a7edbe6d02023c9ad8688c1b12d963361835e139 (patch) | |
tree | ec37ea65114b02cfb4ea5d04a3b653e25c93f573 /src | |
parent | 5bcc0723ff66efd46ad46a16b0d8868557f8e37f (diff) | |
download | armnn-a7edbe6d02023c9ad8688c1b12d963361835e139.tar.gz |
IVGCVSW-6735 DynamicBackendTests throwing exception while running ArmNN Unittest
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I8f3274c39ba34b031f78c5542da6bf1a495892a9
Diffstat (limited to 'src')
-rw-r--r-- | src/dynamic/sample/SampleDynamicLayerSupport.cpp | 25 | ||||
-rw-r--r-- | src/dynamic/sample/SampleDynamicLayerSupport.hpp | 7 |
2 files changed, 32 insertions, 0 deletions
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 |