• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "MaximumLayer.hpp"
7 
8 #include "LayerCloneBase.hpp"
9 
10 #include <armnn/backends/WorkloadData.hpp>
11 #include <armnn/backends/WorkloadFactory.hpp>
12 
13 namespace armnn
14 {
15 
MaximumLayer(const char * name)16 MaximumLayer::MaximumLayer(const char* name)
17     : ElementwiseBaseLayer(2, 1, LayerType::Maximum, name)
18 {
19 }
20 
CreateWorkload(const IWorkloadFactory & factory) const21 std::unique_ptr<IWorkload> MaximumLayer::CreateWorkload(const IWorkloadFactory& factory) const
22 {
23     MaximumQueueDescriptor descriptor;
24     SetAdditionalInfo(descriptor);
25 
26     return factory.CreateWorkload(LayerType::Maximum, descriptor, PrepInfoAndDesc(descriptor));
27 }
28 
Clone(Graph & graph) const29 MaximumLayer* MaximumLayer::Clone(Graph& graph) const
30 {
31     return CloneBase<MaximumLayer>(graph, GetName());
32 }
33 
ExecuteStrategy(IStrategy & strategy) const34 void MaximumLayer::ExecuteStrategy(IStrategy& strategy) const
35 {
36     strategy.ExecuteStrategy(this, GetParameters(), {}, GetName());
37 }
38 
39 } // namespace armnn
40