20.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
armnnUtils::ParserPrototxtFixture< armnnTfParser::ITfParser >::m_Prototext
std::string m_Prototext
Definition:
ParserPrototxtFixture.hpp:68
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:86
armnnUtils::ParserPrototxtFixture
Definition:
ParserPrototxtFixture.hpp:24
src
armnnTfParser
test
Addition.cpp
Generated on Fri Mar 13 2020 16:09:09 for ArmNN by
1.8.13