xref: /aosp_15_r20/external/armnn/src/backends/tosaReference/TosaRefLayerSupport.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include <backendsCommon/LayerSupportBase.hpp>
8 #include <backendsCommon/LayerSupportRules.hpp>
9 
10 namespace armnn {
11 
12 class TosaRefLayerSupport : public ILayerSupport {
13 public:
14     bool IsLayerSupported(const LayerType& type,
15                           const std::vector<TensorInfo>& infos,
16                           const BaseDescriptor& descriptor,
17                           const Optional<LstmInputParamsInfo>& lstmParamsInfo,
18                           const Optional<QuantizedLstmInputParamsInfo>&,
19                           Optional<std::string&> reasonIfUnsupported) const override;
20 
21 };
22 
23 }// namespace armnn
24