Release 18.02
Change-Id: Id3c11dc5ee94ef664374a988fcc6901e9a232fa6
diff --git a/src/armnnCaffeParser/test/TestPooling2d.cpp b/src/armnnCaffeParser/test/TestPooling2d.cpp
new file mode 100644
index 0000000..25cd124
--- /dev/null
+++ b/src/armnnCaffeParser/test/TestPooling2d.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 GlobalPoolingFixture : public ParserPrototxtFixture<armnnCaffeParser::ICaffeParser>
+{
+ GlobalPoolingFixture()
+ {
+ m_Prototext = "name: \"GlobalPooling\"\n"
+ "layer {\n"
+ " name: \"data\"\n"
+ " type: \"Input\"\n"
+ " top: \"data\"\n"
+ " input_param { shape: { dim: 1 dim: 3 dim: 2 dim: 2 } }\n"
+ "}\n"
+ "layer {\n"
+ " bottom: \"data\"\n"
+ " top: \"pool1\"\n"
+ " name: \"pool1\"\n"
+ " type: \"Pooling\"\n"
+ " pooling_param {\n"
+ " pool: AVE\n"
+ " global_pooling: true\n"
+ " }\n"
+ "}\n";
+ SetupSingleInputSingleOutput("data", "pool1");
+ }
+};
+
+BOOST_FIXTURE_TEST_CASE(GlobalPooling, GlobalPoolingFixture)
+{
+ RunTest<4>(
+ {
+ 1,3,
+ 5,7,
+
+ 2,2,
+ 2,2,
+
+ 4,4,
+ 6,6
+ },
+ {
+ 4, 2, 5
+ });
+}
+
+BOOST_AUTO_TEST_SUITE_END()