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