ArmNN  NotReleased
MultiOutput.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 MultiOutMatchFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser>
13 {
14  MultiOutMatchFixture()
15  {
16  m_Prototext = R"(
17 node {
18  name: "input"
19  op: "Placeholder"
20  attr {
21  key: "dtype"
22  value {
23  type: DT_FLOAT
24  }
25  }
26  attr {
27  key: "shape"
28  value {
29  shape {
30  }
31  }
32  }
33 }
34 node {
35  name: "softmax1"
36  op: "Softmax"
37  input: "input:0"
38  attr {
39  key: "T"
40  value {
41  type: DT_FLOAT
42  }
43  }
44 }
45  )";
46  SetupSingleInputSingleOutput({ 1, 7 }, "input", "softmax1");
47  }
48 };
49 
50 BOOST_FIXTURE_TEST_CASE(MultiOutMatch, MultiOutMatchFixture)
51 {
52  // Note that the point of this test is to verify the parsing went well.
53  // Here we make sure the softmax has really connected to the input layer.
54  RunTest<2>({ 0, 0, 10000, 0, 0, 0, 0 }, { 0, 0, 1, 0, 0, 0, 0 });
55 }
56 
57 struct MultiOutFailFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser>
58 {
59  MultiOutFailFixture()
60  {
61  m_Prototext = R"(
62 node {
63  name: "input"
64  op: "Placeholder"
65  attr {
66  key: "dtype"
67  value {
68  type: DT_FLOAT
69  }
70  }
71  attr {
72  key: "shape"
73  value {
74  shape {
75  }
76  }
77  }
78 }
79 node {
80  name: "softmax1"
81  op: "Softmax"
82  input: "input:1"
83  attr {
84  key: "T"
85  value {
86  type: DT_FLOAT
87  }
88  }
89 }
90  )";
91  BOOST_CHECK_THROW(SetupSingleInputSingleOutput({ 1, 7 }, "input", "softmax1"), armnn::ParseException);
92  }
93 };
94 
95 BOOST_FIXTURE_TEST_CASE(MultiOutFail, MultiOutFailFixture)
96 {
97  // Not running the graph because this is expected to throw an exception during parsing.
98 }
99 
100 struct MultiOutInvalidFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser>
101 {
102  MultiOutInvalidFixture()
103  {
104  m_Prototext = R"(
105 node {
106  name: "input"
107  op: "Placeholder"
108  attr {
109  key: "dtype"
110  value {
111  type: DT_FLOAT
112  }
113  }
114  attr {
115  key: "shape"
116  value {
117  shape {
118  }
119  }
120  }
121 }
122 node {
123  name: "softmax1"
124  op: "Softmax"
125  input: "input:-1"
126  attr {
127  key: "T"
128  value {
129  type: DT_FLOAT
130  }
131  }
132 }
133  )";
134  BOOST_CHECK_THROW(SetupSingleInputSingleOutput({ 1, 7 }, "input", "softmax1"), armnn::ParseException);
135  }
136 };
137 
138 BOOST_FIXTURE_TEST_CASE(MultiOutInvalid, MultiOutInvalidFixture)
139 {
140  // Not running the graph because this is expected to throw an exception during parsing.
141 }
142 
143 
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
BOOST_AUTO_TEST_SUITE_END()
BOOST_FIXTURE_TEST_CASE(MultiOutMatch, MultiOutMatchFixture)
Definition: MultiOutput.cpp:50
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)