Narumol Prangnawarat | 867eba5 | 2020-02-03 12:29:56 +0000 | [diff] [blame] | 1 | // |
| 2 | // Copyright © 2020 Arm Ltd. All rights reserved. |
| 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
| 5 | |
| 6 | #pragma once |
| 7 | |
| 8 | #include <backendsCommon/LayerSupportBase.hpp> |
| 9 | |
Jan Eilers | c84e45d | 2020-08-19 14:14:36 +0100 | [diff] [blame] | 10 | namespace sdb // sample dynamic backend |
Narumol Prangnawarat | 867eba5 | 2020-02-03 12:29:56 +0000 | [diff] [blame] | 11 | { |
| 12 | |
Jan Eilers | c84e45d | 2020-08-19 14:14:36 +0100 | [diff] [blame] | 13 | class SampleDynamicLayerSupport : public armnn::LayerSupportBase |
Narumol Prangnawarat | 867eba5 | 2020-02-03 12:29:56 +0000 | [diff] [blame] | 14 | { |
| 15 | public: |
Jan Eilers | c84e45d | 2020-08-19 14:14:36 +0100 | [diff] [blame] | 16 | bool IsAdditionSupported(const armnn::TensorInfo& input0, |
| 17 | const armnn::TensorInfo& input1, |
| 18 | const armnn::TensorInfo& output, |
| 19 | armnn::Optional<std::string&> reasonIfUnsupported = armnn::EmptyOptional()) const override; |
Narumol Prangnawarat | 867eba5 | 2020-02-03 12:29:56 +0000 | [diff] [blame] | 20 | |
Jan Eilers | c84e45d | 2020-08-19 14:14:36 +0100 | [diff] [blame] | 21 | bool IsInputSupported(const armnn::TensorInfo& input, |
| 22 | armnn::Optional<std::string&> reasonIfUnsupported) const override; |
Narumol Prangnawarat | 867eba5 | 2020-02-03 12:29:56 +0000 | [diff] [blame] | 23 | |
Jan Eilers | c84e45d | 2020-08-19 14:14:36 +0100 | [diff] [blame] | 24 | bool IsOutputSupported(const armnn::TensorInfo& output, |
| 25 | armnn::Optional<std::string&> reasonIfUnsupported) const override; |
Cathal Corbett | a7edbe6 | 2022-01-20 15:53:31 +0000 | [diff] [blame] | 26 | |
| 27 | bool IsLayerSupported(const armnn::LayerType& type, |
| 28 | const std::vector<armnn::TensorInfo>& infos, |
| 29 | const armnn::BaseDescriptor& descriptor, |
| 30 | const armnn::Optional<armnn::LstmInputParamsInfo>& lstmParamsInfo, |
| 31 | const armnn::Optional<armnn::QuantizedLstmInputParamsInfo>& quantizedLstmParamsInfo, |
| 32 | armnn::Optional<std::string&> reasonIfUnsupported = armnn::EmptyOptional()) const override; |
Narumol Prangnawarat | 867eba5 | 2020-02-03 12:29:56 +0000 | [diff] [blame] | 33 | }; |
| 34 | |
Jan Eilers | c84e45d | 2020-08-19 14:14:36 +0100 | [diff] [blame] | 35 | } // namespace sdb |