1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #include "ReshapeLayer.hpp"
6
7 #include "LayerCloneBase.hpp"
8
9 #include <armnn/utility/IgnoreUnused.hpp>
10 #include <armnn/TypesUtils.hpp>
11 #include <backendsCommon/WorkloadData.hpp>
12 #include <backendsCommon/WorkloadFactory.hpp>
13
14 namespace armnn
15 {
16
ReshapeLayer(const ReshapeDescriptor & param,const char * name)17 ReshapeLayer::ReshapeLayer(const ReshapeDescriptor& param, const char* name)
18 : LayerWithParameters(1, 1, LayerType::Reshape, param, name)
19 {
20 }
21
CreateWorkload(const IWorkloadFactory & factory) const22 std::unique_ptr<IWorkload> ReshapeLayer::CreateWorkload(const IWorkloadFactory& factory) const
23 {
24 ReshapeQueueDescriptor descriptor;
25 SetAdditionalInfo(descriptor);
26
27 return factory.CreateReshape(descriptor, PrepInfoAndDesc(descriptor));
28 }
29
Clone(Graph & graph) const30 ReshapeLayer* ReshapeLayer::Clone(Graph& graph) const
31 {
32 return CloneBase<ReshapeLayer>(graph, m_Param, GetName());
33 }
34
InferOutputShapes(const std::vector<TensorShape> & inputShapes) const35 std::vector<TensorShape> ReshapeLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) const
36 {
37 IgnoreUnused(inputShapes);
38 return std::vector<TensorShape>({ m_Param.m_TargetShape });
39 }
40
ValidateTensorShapesFromInputs()41 void ReshapeLayer::ValidateTensorShapesFromInputs()
42 {
43 VerifyLayerConnections(1, CHECK_LOCATION());
44
45 const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape();
46
47 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod);
48
49 auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape() });
50
51 ARMNN_ASSERT(inferredShapes.size() == 1);
52
53 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "ReshapeLayer");
54 }
55
Accept(ILayerVisitor & visitor) const56 void ReshapeLayer::Accept(ILayerVisitor& visitor) const
57 {
58 visitor.VisitReshapeLayer(this, GetParameters(), GetName());
59 }
60
61 } // namespace armnn
62