From 6940dd720ebb6b3d1df8ca203ab696daefe58189 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 20 Mar 2020 12:25:56 +0000 Subject: renamed Documentation folder 20.02 and added .nojekyll file Signed-off-by: Jim Flynn --- 20.02/_shape_8cpp_source.xhtml | 121 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 121 insertions(+) create mode 100644 20.02/_shape_8cpp_source.xhtml (limited to '20.02/_shape_8cpp_source.xhtml') diff --git a/20.02/_shape_8cpp_source.xhtml b/20.02/_shape_8cpp_source.xhtml new file mode 100644 index 0000000000..287ca406ad --- /dev/null +++ b/20.02/_shape_8cpp_source.xhtml @@ -0,0 +1,121 @@ + + + + + + + + + + + + + +ArmNN: src/armnnTfParser/test/Shape.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
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>
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 
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
+
BOOST_FIXTURE_TEST_CASE(ParseShape, ShapeFixture)
Definition: Shape.cpp:86
+ + + +
BOOST_AUTO_TEST_SUITE_END()
+
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
Parses and loads the network defined by the m_Prototext string.
+ +
+
+ + + + -- cgit v1.2.1