ArmNN
 21.02
Addition.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 AdditionFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser>
13 {
14  AdditionFixture()
15  {
16  m_Prototext = "node { \n"
17  " name: \"graphInput\" \n"
18  " op: \"Placeholder\" \n"
19  " attr { \n"
20  " key: \"dtype\" \n"
21  " value { \n"
22  " type: DT_FLOAT \n"
23  " } \n"
24  " } \n"
25  " attr { \n"
26  " key: \"shape\" \n"
27  " value { \n"
28  " shape { \n"
29  " } \n"
30  " } \n"
31  " } \n"
32  " } \n"
33  " node { \n"
34  " name: \"softmax1\" \n"
35  " op: \"Softmax\" \n"
36  " input: \"graphInput\" \n"
37  " attr { \n"
38  " key: \"T\" \n"
39  " value { \n"
40  " type: DT_FLOAT \n"
41  " } \n"
42  " } \n"
43  " }\n"
44  " node {\n"
45  " name: \"softmax2\"\n"
46  " op : \"Softmax\"\n"
47  " input: \"graphInput\"\n"
48  " attr { \n"
49  " key: \"T\" \n"
50  " value { \n"
51  " type: DT_FLOAT \n"
52  " } \n"
53  " } \n"
54  " }\n"
55  " node {\n"
56  " name: \"addition\"\n"
57  " op : \"Add\"\n"
58  " input: \"softmax1\"\n"
59  " input: \"softmax2\"\n"
60  " attr { \n"
61  " key: \"T\" \n"
62  " value { \n"
63  " type: DT_FLOAT \n"
64  " } \n"
65  " } \n"
66  " }\n";
67 
68  SetupSingleInputSingleOutput({ 1, 7 }, "graphInput", "addition");
69  }
70 };
71 
72 BOOST_FIXTURE_TEST_CASE(ParseAddition, AdditionFixture)
73 {
74  RunTest<2>({ 0, 0, 10000, 0, 0, 0, 0 }, { 0, 0, 2, 0, 0, 0, 0 });
75 }
76 
77 
78 BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE
BOOST_FIXTURE_TEST_CASE(ValidAddTest, AddValidFixture)
Definition: Addition.cpp:386
ITfParser.hpp
ParserPrototxtFixture.hpp
BOOST_AUTO_TEST_SUITE_END
BOOST_AUTO_TEST_SUITE_END()
armnnUtils::ParserPrototxtFixture::SetupSingleInputSingleOutput
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
Parses and loads the network defined by the m_Prototext string.
Definition: ParserPrototxtFixture.hpp:85