1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #pragma once
7
8 #include <CommonTestUtils.hpp>
9
10 #include <armnn/INetwork.hpp>
11 #include <armnn/TypesUtils.hpp>
12
13 #include <ResolveType.hpp>
14
15 #include <doctest/doctest.h>
16
17 namespace
18 {
19
CreateTransposeNetwork(const armnn::TensorInfo & inputTensorInfo,const armnn::TensorInfo & outputTensorInfo,const armnn::PermutationVector & mappings)20 armnn::INetworkPtr CreateTransposeNetwork(const armnn::TensorInfo& inputTensorInfo,
21 const armnn::TensorInfo& outputTensorInfo,
22 const armnn::PermutationVector& mappings)
23 {
24 armnn::INetworkPtr network(armnn::INetwork::Create());
25
26 const armnn::TransposeDescriptor transposeDescriptor(mappings);
27
28 armnn::IConnectableLayer* inputLayer = network->AddInputLayer(0, "Input");
29 armnn::IConnectableLayer* transposeLayer = network->AddTransposeLayer(transposeDescriptor, "Transpose");
30 armnn::IConnectableLayer* outputLayer = network->AddOutputLayer(0, "Output");
31
32 Connect(inputLayer, transposeLayer, inputTensorInfo, 0, 0);
33 Connect(transposeLayer, outputLayer, outputTensorInfo, 0, 0);
34
35 return network;
36 }
37
38 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
TransposeEndToEnd(const std::vector<armnn::BackendId> & backends)39 void TransposeEndToEnd(const std::vector<armnn::BackendId>& backends)
40 {
41 using namespace armnn;
42
43 std::vector<float> floatInputData{
44 1, 2, 3, 4, 5,
45 11, 12, 13, 14, 15,
46 21, 22, 23, 24, 25
47 };
48 std::vector<T> inputData = armnnUtils::QuantizedVector<T>(floatInputData);
49 std::vector<T> expectedOutputData = armnnUtils::QuantizedVector<T>(floatInputData);
50
51 const armnn::PermutationVector mappings{0, 2, 1 ,3};
52
53 TensorInfo inputInfo ({ 1, 1, 5, 3 }, ArmnnType, 0.0f, 0, true);
54 TensorInfo outputInfo({ 1, 5, 1, 3 }, ArmnnType, 0.0f, 0, true);
55
56 armnn::INetworkPtr network = CreateTransposeNetwork(inputInfo, outputInfo, mappings);
57
58 CHECK(network);
59
60 std::map<int, std::vector<T>> inputTensorData = {{ 0, inputData }};
61 std::map<int, std::vector<T>> expectedOutputTensorData = {{ 0, expectedOutputData }};
62
63 EndToEndLayerTestImpl<ArmnnType, DataType::Signed32>(std::move(network),
64 inputTensorData,
65 expectedOutputTensorData,
66 backends);
67 }
68
69 } // anonymous namespace