diff options
Diffstat (limited to 'src/dynamic/sample/SampleDynamicLayerSupport.hpp')
-rw-r--r-- | src/dynamic/sample/SampleDynamicLayerSupport.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
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 |