2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
5 #include "SoftmaxLayer.hpp"
7 #include "LayerCloneBase.hpp"
9 #include <armnn/TypesUtils.hpp>
10 #include <backendsCommon/WorkloadData.hpp>
11 #include <backendsCommon/WorkloadFactory.hpp>
16 SoftmaxLayer::SoftmaxLayer(const SoftmaxDescriptor ¶m, const char* name)
17 : LayerWithParameters(1, 1, LayerType::Softmax, param, name)
21 std::unique_ptr<IWorkload> SoftmaxLayer::CreateWorkload(const Graph& graph, const IWorkloadFactory& factory) const
23 SoftmaxQueueDescriptor descriptor;
24 return factory.CreateSoftmax(descriptor, PrepInfoAndDesc(descriptor, graph));
27 SoftmaxLayer* SoftmaxLayer::Clone(Graph& graph) const
29 return CloneBase<SoftmaxLayer>(graph, m_Param, GetName());
32 void SoftmaxLayer::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 "SoftmaxLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.",
42 GetOutputSlot(0).GetTensorInfo().GetShape(),