1 //
2 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "TransposeLayer.hpp"
7
8 #include "LayerCloneBase.hpp"
9
10 #include <armnn/TypesUtils.hpp>
11
12 #include <armnnUtils/Transpose.hpp>
13
14 #include <armnn/backends/WorkloadData.hpp>
15 #include <armnn/backends/WorkloadFactory.hpp>
16
17 namespace armnn
18 {
19
TransposeLayer(const TransposeDescriptor & param,const char * name)20 TransposeLayer::TransposeLayer(const TransposeDescriptor& param, const char* name)
21 : LayerWithParameters(1, 1, LayerType::Transpose, param, name)
22 {
23 }
24
CreateWorkload(const IWorkloadFactory & factory) const25 std::unique_ptr<IWorkload> TransposeLayer::CreateWorkload(const IWorkloadFactory& factory) const
26 {
27 TransposeQueueDescriptor descriptor;
28 SetAdditionalInfo(descriptor);
29
30 return factory.CreateWorkload(LayerType::Transpose, descriptor, PrepInfoAndDesc(descriptor));
31 }
32
Clone(Graph & graph) const33 TransposeLayer* TransposeLayer::Clone(Graph& graph) const
34 {
35 return CloneBase<TransposeLayer>(graph, m_Param, GetName());
36 }
37
InferOutputShapes(const std::vector<TensorShape> & inputShapes) const38 std::vector<TensorShape> TransposeLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) const
39 {
40 ARMNN_ASSERT(inputShapes.size() == 1);
41 const TensorShape& inShape = inputShapes[0];
42 return std::vector<TensorShape> ({armnnUtils::TransposeTensorShape(inShape, m_Param.m_DimMappings)});
43 }
44
ValidateTensorShapesFromInputs()45 void TransposeLayer::ValidateTensorShapesFromInputs()
46 {
47 VerifyLayerConnections(1, CHECK_LOCATION());
48
49 const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape();
50
51 VerifyShapeInferenceType(outputShape, m_ShapeInferenceMethod);
52
53 auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape() });
54
55 ARMNN_ASSERT(inferredShapes.size() == 1);
56
57 ValidateAndCopyShape(outputShape, inferredShapes[0], m_ShapeInferenceMethod, "TransposeLayer");
58 }
59
ExecuteStrategy(IStrategy & strategy) const60 void TransposeLayer::ExecuteStrategy(IStrategy& strategy) const
61 {
62 strategy.ExecuteStrategy(this, GetParameters(), {}, GetName());
63 }
64
65 } // namespace armnn
66