aboutsummaryrefslogtreecommitdiff
path: root/src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp')
-rw-r--r--src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp54
1 files changed, 54 insertions, 0 deletions
diff --git a/src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp b/src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
new file mode 100644
index 0000000000..cd87246bee
--- /dev/null
+++ b/src/armnnCaffeParser/test/TestMultiInputsOutputs.cpp
@@ -0,0 +1,54 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// See LICENSE file in the project root for full license information.
+//
+#include <boost/test/unit_test.hpp>
+#include "armnnCaffeParser/ICaffeParser.hpp"
+#include "ParserPrototxtFixture.hpp"
+
+BOOST_AUTO_TEST_SUITE(CaffeParser)
+
+struct MultiInputsOutputsFixture : public ParserPrototxtFixture<armnnCaffeParser::ICaffeParser>
+{
+ MultiInputsOutputsFixture()
+ {
+ m_Prototext = R"(
+name: "MultiInputsOutputs"
+layer {
+ name: "input1"
+ type: "Input"
+ top: "input1"
+ input_param { shape: { dim: 1 } }
+}
+layer {
+ name: "input2"
+ type: "Input"
+ top: "input2"
+ input_param { shape: { dim: 1 } }
+}
+layer {
+ bottom: "input1"
+ bottom: "input2"
+ top: "add1"
+ name: "add1"
+ type: "Eltwise"
+}
+layer {
+ bottom: "input2"
+ bottom: "input1"
+ top: "add2"
+ name: "add2"
+ type: "Eltwise"
+}
+ )";
+ Setup({ }, { "add1", "add2" });
+ }
+};
+
+BOOST_FIXTURE_TEST_CASE(MultiInputsOutputs, MultiInputsOutputsFixture)
+{
+ RunTest<1>({ { "input1",{ 12.0f } },{ "input2",{ 13.0f } } },
+ { { "add1",{ 25.0f } },{ "add2",{ 25.0f } } });
+}
+
+BOOST_AUTO_TEST_SUITE_END()