2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
5 #include "NormalizationLayer.hpp"
7 #include "LayerCloneBase.hpp"
9 #include <armnn/TypesUtils.hpp>
10 #include <backendsCommon/WorkloadData.hpp>
11 #include <backendsCommon/WorkloadFactory.hpp>
16 NormalizationLayer::NormalizationLayer(const NormalizationDescriptor& param, const char* name)
17 : LayerWithParameters(1, 1, LayerType::Normalization, param, name)
21 std::unique_ptr<IWorkload> NormalizationLayer::CreateWorkload(const Graph& graph, const IWorkloadFactory& factory) const
23 NormalizationQueueDescriptor descriptor;
24 return factory.CreateNormalization(descriptor, PrepInfoAndDesc(descriptor, graph));
27 NormalizationLayer* NormalizationLayer::Clone(Graph& graph) const
29 return CloneBase<NormalizationLayer>(graph, m_Param, GetName());
32 void NormalizationLayer::ValidateTensorShapesFromInputs()
34 VerifyLayerConnections(1, CHECK_LOCATION());
36 auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape() });
38 BOOST_ASSERT(inferredShapes.size() == 1);
40 ConditionalThrowIfNotEqual<LayerValidationException>(
41 "NormalizationLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.",
42 GetOutputSlot(0).GetTensorInfo().GetShape(),