1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "MinimumLayer.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 MinimumLayer(const char * name)17MinimumLayer::MinimumLayer(const char* name) 18 : ElementwiseBaseLayer(2, 1, LayerType::Minimum, name) 19 { 20 } 21 CreateWorkload(const IWorkloadFactory & factory) const22std::unique_ptr<IWorkload> MinimumLayer::CreateWorkload(const IWorkloadFactory& factory) const 23 { 24 MinimumQueueDescriptor descriptor; 25 SetAdditionalInfo(descriptor); 26 27 return factory.CreateMinimum(descriptor, PrepInfoAndDesc(descriptor)); 28 } 29 Clone(Graph & graph) const30MinimumLayer* MinimumLayer::Clone(Graph& graph) const 31 { 32 return CloneBase<MinimumLayer>(graph, GetName()); 33 } 34 Accept(ILayerVisitor & visitor) const35void MinimumLayer::Accept(ILayerVisitor& visitor) const 36 { 37 visitor.VisitMinimumLayer(this, GetName()); 38 } 39 40 } // namespace armnn 41