src/armnnTfParser/test/Addition.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // See LICENSE file in the project root for full license information.
   4 //
   5
   6 #include <boost/test/unit_test.hpp>
   7 #include "armnnTfParser/ITfParser.hpp"
   8 #include "ParserPrototxtFixture.hpp"
   9
  10 BOOST_AUTO_TEST_SUITE(TensorflowParser)
  11
  12 struct AdditionFixture : public ParserPrototxtFixture<armnnTfParser::ITfParser>
  13 {
  14     AdditionFixture()
  15     {
  16         m_Prototext = "node { \n"
  17             "    name: \"graphInput\" \n"
  18             "    op: \"Placeholder\" \n"
  19             "    attr { \n"
  20             "      key: \"dtype\" \n"
  21             "      value { \n"
  22             "        type: DT_FLOAT \n"
  23             "      } \n"
  24             "    } \n"
  25             "    attr { \n"
  26             "      key: \"shape\" \n"
  27             "      value { \n"
  28             "        shape { \n"
  29             "        } \n"
  30             "      } \n"
  31             "    } \n"
  32             "  } \n"
  33             "  node { \n"
  34             "    name: \"softmax1\" \n"
  35             "    op: \"Softmax\" \n"
  36             "    input: \"graphInput\" \n"
  37             "    attr { \n"
  38             "      key: \"T\" \n"
  39             "      value { \n"
  40             "        type: DT_FLOAT \n"
  41             "      } \n"
  42             "    } \n"
  43             "  }\n"
  44             "  node {\n"
  45             "    name: \"softmax2\"\n"
  46             "    op : \"Softmax\"\n"
  47             "    input: \"graphInput\"\n"
  48             "    attr { \n"
  49             "      key: \"T\" \n"
  50             "      value { \n"
  51             "        type: DT_FLOAT \n"
  52             "      } \n"
  53             "    } \n"
  54             "  }\n"
  55             "  node {\n"
  56             "    name: \"addition\"\n"
  57             "    op : \"Add\"\n"
  58             "    input: \"softmax1\"\n"
  59             "    input: \"softmax2\"\n"
  60             "    attr { \n"
  61             "      key: \"T\" \n"
  62             "      value { \n"
  63             "        type: DT_FLOAT \n"
  64             "      } \n"
  65             "    } \n"
  66             "  }\n";
  67
  68         SetupSingleInputSingleOutput({ 1, 7 }, "graphInput", "addition");
  69     }
  70 };
  71
  72 BOOST_FIXTURE_TEST_CASE(ParseAddition, AdditionFixture)
  73 {
  74     RunTest<2>({ 0, 0, 10000, 0, 0, 0, 0 }, { 0, 0, 2, 0, 0, 0, 0 });
  75 }
  76
  77
  78 BOOST_AUTO_TEST_SUITE_END()