src/armnnCaffeParser/test/TestConcat.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // See LICENSE file in the project root for full license information.
   4 //
   5 #include <boost/test/unit_test.hpp>
   6 #include "armnnCaffeParser/ICaffeParser.hpp"
   7 #include "ParserPrototxtFixture.hpp"
   8
   9 BOOST_AUTO_TEST_SUITE(CaffeParser)
  10
  11 struct ConcatFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser>
  12 {
  13     ConcatFixture()
  14     {
  15         m_Prototext = "name: \"Concat\"\n"
  16             "layer {\n"
  17             "  name: \"data\"\n"
  18             "  type: \"Input\"\n"
  19             "  top: \"data\"\n"
  20             "  input_param { shape: { dim: 1 dim: 1 dim: 4 dim: 4 } }\n"
  21             "}\n"
  22             "layer {\n"
  23             "    bottom: \"data\"\n"
  24             "    top: \"pool1\"\n"
  25             "    name: \"pool1\"\n"
  26             "    type: \"Pooling\"\n"
  27             "    pooling_param {\n"
  28             "        kernel_size: 2\n"
  29             "        stride: 2\n"
  30             "        pool: MAX\n"
  31             "    }\n"
  32             "}\n"
  33             "layer {\n"
  34             "    bottom: \"data\"\n"
  35             "    top: \"pool2\"\n"
  36             "    name: \"pool2\"\n"
  37             "    type: \"Pooling\"\n"
  38             "    pooling_param {\n"
  39             "        kernel_size: 2\n"
  40             "        stride: 2\n"
  41             "        pool: MAX\n"
  42             "    }\n"
  43             "}\n"
  44             "layer {\n"
  45             "    bottom: \"pool1\"\n"
  46             "    bottom: \"pool2\"\n"
  47             "    top: \"concat\"\n"
  48             "    name: \"concat\"\n"
  49             "    type: \"Concat\"\n"
  50             "}\n";
  51         SetupSingleInputSingleOutput("data", "concat");
  52     }
  53 };
  54
  55 BOOST_FIXTURE_TEST_CASE(ParseConcat, ConcatFixture)
  56 {
  57     RunTest<4>(
  58         {
  59             0, 1, 0, 0,
  60             0, 0, 0, 0,
  61             0, 0, 1, 0,
  62             1, 0, 1, 1
  63         },
  64         {
  65             1, 0,
  66             1, 1,
  67
  68             1, 0,
  69             1, 1
  70         });
  71 }
  72
  73 BOOST_AUTO_TEST_SUITE_END()