ArmNN
 24.02
FusedLayer.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2023 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
10 
11 #include <armnn/Descriptors.hpp>
12 
13 #include <memory>
14 #include <functional>
15 
16 namespace armnn
17 {
18 
19 class FusedLayer : public LayerWithParameters<FusedDescriptor>
20 {
21 public:
22  FusedLayer(const FusedDescriptor& param, const char* name);
23  ~FusedLayer();
24 
25  virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override;
26 
27  FusedLayer* Clone(Graph &graph) const override;
28 
29  void ValidateTensorShapesFromInputs() override;
30 
31  void ExecuteStrategy(IStrategy& strategy) const override;
32 
33 private:
34  FusedLayer(const FusedLayer& other) = delete;
35  FusedLayer& operator=(const FusedLayer& other) = delete;
36 };
37 
38 } // namespace armnn
Descriptors.hpp
armnn::FusedLayer
Definition: FusedLayer.hpp:19
armnn::FusedLayer::ExecuteStrategy
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
Definition: FusedLayer.cpp:43
armnn::IStrategy
Definition: IStrategy.hpp:16
WorkloadFactory.hpp
armnn::LayerWithParameters
Definition: LayerWithParameters.hpp:14
armnn::FusedDescriptor
A FusedDescriptor for the FusedLayer.
Definition: Descriptors.hpp:944
armnn::FusedLayer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Definition: FusedLayer.cpp:29
armnn::IWorkloadFactory
Definition: WorkloadFactory.hpp:22
armnn::FusedLayer::~FusedLayer
~FusedLayer()
Definition: FusedLayer.cpp:19
armnn::FusedLayer::Clone
FusedLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
Definition: FusedLayer.cpp:22
LayerWithParameters.hpp
armnn::FusedLayer::FusedLayer
FusedLayer(const FusedDescriptor &param, const char *name)
Definition: FusedLayer.cpp:15
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::FusedLayer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Definition: FusedLayer.cpp:37
armnn::Graph
Definition: Graph.hpp:30