Release 18.02
Change-Id: Id3c11dc5ee94ef664374a988fcc6901e9a232fa6
diff --git a/src/armnnCaffeParser/test/TestDropout.cpp b/src/armnnCaffeParser/test/TestDropout.cpp
new file mode 100644
index 0000000..16f2c27
--- /dev/null
+++ b/src/armnnCaffeParser/test/TestDropout.cpp
@@ -0,0 +1,53 @@
+//
+// 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 DropoutFixture : public ParserPrototxtFixture<armnnCaffeParser::ICaffeParser>
+{
+ DropoutFixture()
+ {
+ m_Prototext = "name: \"DropoutFixture\"\n"
+ "layer {\n"
+ " name: \"data\"\n"
+ " type: \"Input\"\n"
+ " top: \"data\"\n"
+ " input_param { shape: { dim: 1 dim: 1 dim: 2 dim: 2 } }\n"
+ "}\n"
+ "layer {\n"
+ " bottom: \"data\"\n"
+ " top: \"drop1\"\n"
+ " name: \"drop1\"\n"
+ " type: \"Dropout\"\n"
+ "}\n"
+ "layer {\n"
+ " bottom: \"drop1\"\n"
+ " bottom: \"drop1\"\n"
+ " top: \"add\"\n"
+ " name: \"add\"\n"
+ " type: \"Eltwise\"\n"
+ "}\n";
+ SetupSingleInputSingleOutput("data", "add");
+ }
+};
+
+BOOST_FIXTURE_TEST_CASE(ParseDropout, DropoutFixture)
+{
+ RunTest<4>(
+ {
+ 1, 2,
+ 3, 4,
+ },
+ {
+ 2, 4,
+ 6, 8
+ });
+}
+
+BOOST_AUTO_TEST_SUITE_END()