ArmNN
NotReleased
Shape.cpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#include <boost/test/unit_test.hpp>
7
#include "
armnnTfParser/ITfParser.hpp
"
8
#include "
ParserPrototxtFixture.hpp
"
9
10
BOOST_AUTO_TEST_SUITE
(TensorflowParser)
11
12
struct
ShapeFixture :
public
armnnUtils::ParserPrototxtFixture
<armnnTfParser::ITfParser>
13
{
14
ShapeFixture()
15
{
16
m_Prototext
=
17
"node { \n"
18
" name: \"Placeholder\" \n"
19
" op: \"Placeholder\" \n"
20
" attr { \n"
21
" key: \"dtype\" \n"
22
" value { \n"
23
" type: DT_FLOAT \n"
24
" } \n"
25
" } \n"
26
" attr { \n"
27
" key: \"shape\" \n"
28
" value { \n"
29
" shape { \n"
30
" dim { \n"
31
" size: 1 \n"
32
" } \n"
33
" dim { \n"
34
" size: 1 \n"
35
" } \n"
36
" dim { \n"
37
" size: 1 \n"
38
" } \n"
39
" dim { \n"
40
" size: 4 \n"
41
" } \n"
42
" } \n"
43
" } \n"
44
" } \n"
45
"} \n"
46
"node { \n"
47
" name: \"shapeTest\" \n"
48
" op: \"Shape\" \n"
49
" input: \"Placeholder\" \n"
50
" attr { \n"
51
" key: \"T\" \n"
52
" value { \n"
53
" type: DT_FLOAT \n"
54
" } \n"
55
" } \n"
56
" attr { \n"
57
" key: \"out_type\" \n"
58
" value { \n"
59
" type: DT_INT32 \n"
60
" } \n"
61
" } \n"
62
"} \n"
63
"node { \n"
64
" name: \"Reshape\" \n"
65
" op: \"Reshape\" \n"
66
" input: \"Placeholder\" \n"
67
" input: \"shapeTest\" \n"
68
" attr { \n"
69
" key: \"T\" \n"
70
" value { \n"
71
" type: DT_FLOAT \n"
72
" } \n"
73
" } \n"
74
" attr { \n"
75
" key: \"Tshape\" \n"
76
" value { \n"
77
" type: DT_INT32 \n"
78
" } \n"
79
" } \n"
80
"} \n"
;
81
82
SetupSingleInputSingleOutput
({1, 4},
"Placeholder"
,
"Reshape"
);
83
}
84
};
85
86
BOOST_FIXTURE_TEST_CASE
(ParseShape, ShapeFixture)
87
{
88
// Note: the test's output cannot be an int32 const layer, because ARMNN only supports u8 and float layers.
89
// For that reason I added a reshape layer which reshapes the input to its original dimensions.
90
RunTest<2>({ 0.0f, 1.0f, 2.0f, 3.0f }, { 0.0f, 1.0f, 2.0f, 3.0f });
91
}
92
93
BOOST_AUTO_TEST_SUITE_END
()
armnnUtils::ParserPrototxtFixture::SetupSingleInputSingleOutput
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
Definition:
ParserPrototxtFixture.hpp:86
armnnUtils::ParserPrototxtFixture< armnnTfParser::ITfParser >::m_Prototext
std::string m_Prototext
Definition:
ParserPrototxtFixture.hpp:68
BOOST_FIXTURE_TEST_CASE
BOOST_FIXTURE_TEST_CASE(ParseShape, ShapeFixture)
Definition:
Shape.cpp:86
BOOST_AUTO_TEST_SUITE_END
BOOST_AUTO_TEST_SUITE_END()
armnnUtils::ParserPrototxtFixture
Definition:
ParserPrototxtFixture.hpp:24
BOOST_AUTO_TEST_SUITE
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
ParserPrototxtFixture.hpp
ITfParser.hpp
src
armnnTfParser
test
Shape.cpp
Generated on Fri Mar 13 2020 16:06:56 for ArmNN by
1.8.13