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