src/armnnTfLiteParser/test/Multiplication.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #include <boost/test/unit_test.hpp>
   7 #include "ParserFlatbuffersFixture.hpp"
   8 #include "../TfLiteParser.hpp"
   9
  10 #include <string>
  11 #include <iostream>
  12
  13 BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
  14
  15 struct MultiplicationFixture : public ParserFlatbuffersFixture
  16 {
  17     explicit MultiplicationFixture(const std::string & inputShape1,
  18                                             const std::string & inputShape2,
  19                                             const std::string & outputShape,
  20                                             const std::string & activation="NONE")
  21     {
  22         m_JsonString = R"(
  23             {
  24                 "version": 3,
  25                 "operator_codes": [ { "builtin_code": "MUL" } ],
  26                 "subgraphs": [ {
  27                     "tensors": [
  28                         {
  29                             "shape": )" + inputShape1 + R"(,
  30                             "type": "FLOAT32",
  31                             "buffer": 0,
  32                             "name": "inputTensor1",
  33                             "quantization": {
  34                                 "min": [ 0.0 ],
  35                                 "max": [ 255.0 ],
  36                                 "scale": [ 1.0 ],
  37                                 "zero_point": [ 0 ],
  38                             }
  39                         },
  40                         {
  41                             "shape": )" + inputShape2 + R"(,
  42                             "type": "FLOAT32",
  43                             "buffer": 1,
  44                             "name": "inputTensor2",
  45                             "quantization": {
  46                                 "min": [ 0.0 ],
  47                                 "max": [ 255.0 ],
  48                                 "scale": [ 1.0 ],
  49                                 "zero_point": [ 0 ],
  50                             }
  51                         },
  52                         {
  53                             "shape": )" + outputShape + R"( ,
  54                             "type": "FLOAT32",
  55                             "buffer": 2,
  56                             "name": "outputTensor",
  57                             "quantization": {
  58                                 "min": [ 0.0 ],
  59                                 "max": [ 255.0 ],
  60                                 "scale": [ 1.0 ],
  61                                 "zero_point": [ 0 ],
  62                             }
  63                         }
  64                     ],
  65                     "inputs": [ 0, 1 ],
  66                     "outputs": [ 2 ],
  67                     "operators": [
  68                         {
  69                             "opcode_index": 0,
  70                             "inputs": [ 0, 1 ],
  71                             "outputs": [ 2 ],
  72                             "builtin_options_type": "MulOptions",
  73                             "builtin_options": {
  74                                 "fused_activation_function": )" + activation + R"(
  75                             },
  76                             "custom_options_format": "FLEXBUFFERS"
  77                         }
  78                     ],
  79                 } ],
  80                 "buffers" : [
  81                     { },
  82                     { }
  83                 ]
  84             }
  85         )";
  86         Setup();
  87     }
  88 };
  89
  90 struct SimpleMultiplicationFixture : public MultiplicationFixture
  91 {
  92     SimpleMultiplicationFixture() : MultiplicationFixture("[ 1, 2, 2, 3 ]", "[ 1, 2, 2, 3 ]", "[ 1, 2, 2, 3 ]") {}
  93 };
  94
  95 BOOST_FIXTURE_TEST_CASE(ParseMultiplication, SimpleMultiplicationFixture)
  96 {
  97     RunTest<4, armnn::DataType::Float32>(0, {{ "inputTensor1", { 0.0f,  1.0f,  2.0f,
  98                                                                  3.0f,  4.0f,  5.0f,
  99                                                                  6.0f,  7.0f,  8.0f,
 100                                                                  9.0f, 10.0f, 11.0f } },
 101                                              { "inputTensor2", { 1.0f,  1.0f,  1.0f,
 102                                                                  5.0f,  5.0f,  5.0f,
 103                                                                  1.0f,  1.0f,  1.0f,
 104                                                                  5.0f,  5.0f,  5.0f} } },
 105                                              {{ "outputTensor", { 0.0f,  1.0f,  2.0f,
 106                                                                  15.0f, 20.0f, 25.0f,
 107                                                                   6.0f,  7.0f,  8.0f,
 108                                                                  45.0f, 50.0f, 55.0f } } });
 109 }
 110
 111 BOOST_AUTO_TEST_SUITE_END()