1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "MultiplicationLayer.hpp" 7 8 #include "LayerCloneBase.hpp" 9 10 #include <armnn/TypesUtils.hpp> 11 #include <backendsCommon/WorkloadData.hpp> 12 #include <backendsCommon/WorkloadFactory.hpp> 13 14 namespace armnn 15 { 16 MultiplicationLayer(const char * name)17MultiplicationLayer::MultiplicationLayer(const char* name) 18 : ElementwiseBaseLayer(2, 1, LayerType::Multiplication, name) 19 { 20 } 21 CreateWorkload(const IWorkloadFactory & factory) const22std::unique_ptr<IWorkload> MultiplicationLayer::CreateWorkload(const IWorkloadFactory& factory) const 23 { 24 MultiplicationQueueDescriptor descriptor; 25 SetAdditionalInfo(descriptor); 26 27 return factory.CreateMultiplication(descriptor, PrepInfoAndDesc(descriptor)); 28 } 29 Clone(Graph & graph) const30MultiplicationLayer* MultiplicationLayer::Clone(Graph& graph) const 31 { 32 return CloneBase<MultiplicationLayer>(graph, GetName()); 33 } 34 Accept(ILayerVisitor & visitor) const35void MultiplicationLayer::Accept(ILayerVisitor& visitor) const 36 { 37 visitor.VisitMultiplicationLayer(this, GetName()); 38 } 39 40 } // namespace armnn 41