ngraph/test/backend/power.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}, power)
  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::Power>(A, B), ParameterVector{A, B});
  54
  55     std::vector<float> a{1, 2, 3, 5};
  56     std::vector<float> b{2, 0, 6, 3};
  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, {1, 1, 729, 125});
  61     test_case.run();
  62 }