src/armnnCaffeParser/test/TestAdd.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 AddFixture : public armnnUtils::ParserPrototxtFixture<armnnCaffeParser::ICaffeParser>
  12 {
  13     AddFixture()
  14     {
  15         m_Prototext = "name: \"MinimalAdd\"\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: \"add\"\n"
  48                       "    name: \"add\"\n"
  49                       "    type: \"Eltwise\"\n"
  50                       "}\n";
  51         SetupSingleInputSingleOutput("data", "add");
  52     }
  53 };
  54
  55 BOOST_FIXTURE_TEST_CASE(ParseAdd, AddFixture)
  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             2, 0,
  66             2, 2
  67         });
  68 }
  69
  70 BOOST_AUTO_TEST_SUITE_END()