2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
5 #include "AdditionLayer.hpp"
7 #include "LayerCloneBase.hpp"
9 #include <armnn/TypesUtils.hpp>
10 #include <backendsCommon/WorkloadData.hpp>
11 #include <backendsCommon/WorkloadFactory.hpp>
16 AdditionLayer::AdditionLayer(const char* name)
17 : ArithmeticBaseLayer(2, 1, LayerType::Addition, name)
21 std::unique_ptr<IWorkload> AdditionLayer::CreateWorkload(const Graph& graph,
22 const IWorkloadFactory& factory) const
24 AdditionQueueDescriptor descriptor;
25 return factory.CreateAddition(descriptor, PrepInfoAndDesc(descriptor, graph));
28 AdditionLayer* AdditionLayer::Clone(Graph& graph) const
30 return CloneBase<AdditionLayer>(graph, GetName());