1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "../ParserHelper.hpp"
#include <boost/test/unit_test.hpp>
#include "armnn/Types.hpp"
using namespace armnn;
using namespace armnnUtils;
BOOST_AUTO_TEST_SUITE(ParserHelperSuite)
BOOST_AUTO_TEST_CASE(CalculateReducedOutputTensoInfoTest)
{
bool keepDims = false;
unsigned int inputShape[] = { 2, 3, 4 };
TensorInfo inputTensorInfo(3, &inputShape[0], DataType::Float32);
// Reducing all dimensions results in one single output value (one dimension)
std::set<unsigned int> axisData1 = { 0, 1, 2 };
TensorInfo outputTensorInfo1;
CalculateReducedOutputTensoInfo(inputTensorInfo, axisData1, keepDims, outputTensorInfo1);
BOOST_ASSERT(outputTensorInfo1.GetNumDimensions() == 1);
BOOST_ASSERT(outputTensorInfo1.GetShape()[0] == 1);
// Reducing dimension 0 results in a 3x4 size tensor (one dimension)
std::set<unsigned int> axisData2 = { 0 };
TensorInfo outputTensorInfo2;
CalculateReducedOutputTensoInfo(inputTensorInfo, axisData2, keepDims, outputTensorInfo2);
BOOST_ASSERT(outputTensorInfo2.GetNumDimensions() == 1);
BOOST_ASSERT(outputTensorInfo2.GetShape()[0] == 12);
// Reducing dimensions 0,1 results in a 4 size tensor (one dimension)
std::set<unsigned int> axisData3 = { 0, 1 };
TensorInfo outputTensorInfo3;
CalculateReducedOutputTensoInfo(inputTensorInfo, axisData3, keepDims, outputTensorInfo3);
BOOST_ASSERT(outputTensorInfo3.GetNumDimensions() == 1);
BOOST_ASSERT(outputTensorInfo3.GetShape()[0] == 4);
// Reducing dimension 0 results in a { 1, 3, 4 } dimension tensor
keepDims = true;
std::set<unsigned int> axisData4 = { 0 };
TensorInfo outputTensorInfo4;
CalculateReducedOutputTensoInfo(inputTensorInfo, axisData4, keepDims, outputTensorInfo4);
BOOST_ASSERT(outputTensorInfo4.GetNumDimensions() == 3);
BOOST_ASSERT(outputTensorInfo4.GetShape()[0] == 1);
BOOST_ASSERT(outputTensorInfo4.GetShape()[1] == 3);
BOOST_ASSERT(outputTensorInfo4.GetShape()[2] == 4);
// Reducing dimension 1, 2 results in a { 2, 1, 1 } dimension tensor
keepDims = true;
std::set<unsigned int> axisData5 = { 1, 2 };
TensorInfo outputTensorInfo5;
CalculateReducedOutputTensoInfo(inputTensorInfo, axisData5, keepDims, outputTensorInfo5);
BOOST_ASSERT(outputTensorInfo5.GetNumDimensions() == 3);
BOOST_ASSERT(outputTensorInfo5.GetShape()[0] == 2);
BOOST_ASSERT(outputTensorInfo5.GetShape()[1] == 1);
BOOST_ASSERT(outputTensorInfo5.GetShape()[2] == 1);
}
BOOST_AUTO_TEST_SUITE_END()
|
