ngraph/test/backend/multiply.in.cpp

   1 //*****************************************************************************
   2 // Copyright 2017-2020 Intel Corporation
   3 //
   4 // Licensed under the Apache License, Version 2.0 (the "License");
   5 // you may not use this file except in compliance with the License.
   6 // You may obtain a copy of the License at
   7 //
   8 //     http://www.apache.org/licenses/LICENSE-2.0
   9 //
  10 // Unless required by applicable law or agreed to in writing, software
  11 // distributed under the License is distributed on an "AS IS" BASIS,
  12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 // See the License for the specific language governing permissions and
  14 // limitations under the License.
  15 //*****************************************************************************
  16
  17 #include <algorithm>
  18 #include <cinttypes>
  19 #include <cmath>
  20 #include <cstdlib>
  21 #include <random>
  22 #include <string>
  23
  24 // clang-format off
  25 #ifdef ${BACKEND_NAME}_FLOAT_TOLERANCE_BITS
  26 #define DEFAULT_FLOAT_TOLERANCE_BITS ${BACKEND_NAME}_FLOAT_TOLERANCE_BITS
  27 #endif
  28
  29 #ifdef ${BACKEND_NAME}_DOUBLE_TOLERANCE_BITS
  30 #define DEFAULT_DOUBLE_TOLERANCE_BITS ${BACKEND_NAME}_DOUBLE_TOLERANCE_BITS
  31 #endif
  32 // clang-format on
  33
  34 #include "gtest/gtest.h"
  35 #include "ngraph/ngraph.hpp"
  36 #include "util/engine/test_engines.hpp"
  37 #include "util/test_case.hpp"
  38 #include "util/test_control.hpp"
  39
  40 NGRAPH_SUPPRESS_DEPRECATED_START
  41
  42 using namespace std;
  43 using namespace ngraph;
  44
  45 static string s_manifest = "${MANIFEST}";
  46 using TestEngine = test::ENGINE_CLASS_NAME(${BACKEND_NAME});
  47
  48 NGRAPH_TEST(${BACKEND_NAME}, multiply)
  49 {
  50     Shape shape{2, 2};
  51     auto A = make_shared<op::Parameter>(element::f32, shape);
  52     auto B = make_shared<op::Parameter>(element::f32, shape);
  53     auto f = make_shared<Function>(make_shared<op::Multiply>(A, B), ParameterVector{A, B});
  54
  55     std::vector<float> a{1, 2, 3, 4};
  56     std::vector<float> b{5, 6, 7, 8};
  57
  58     auto test_case = test::TestCase<TestEngine>(f);
  59     test_case.add_multiple_inputs<float>({a, b});
  60     test_case.add_expected_output<float>(shape, {5, 12, 21, 32});
  61     test_case.run();
  62 }
  63
  64 NGRAPH_TEST(${BACKEND_NAME}, multiply_overload)
  65 {
  66     Shape shape{2, 2};
  67     auto A = make_shared<op::Parameter>(element::f32, shape);
  68     auto B = make_shared<op::Parameter>(element::f32, shape);
  69     auto f = make_shared<Function>(A * B, ParameterVector{A, B});
  70
  71     std::vector<float> a{1, 2, 3, 4};
  72     std::vector<float> b{5, 6, 7, 8};
  73
  74     auto test_case = test::TestCase<TestEngine>(f);
  75     test_case.add_multiple_inputs<float>({a, b});
  76     test_case.add_expected_output<float>(shape, {5, 12, 21, 32});
  77     test_case.run();
  78 }