Fixed type of reduction axes (#1190)

diff --git a/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/lrn.cpp b/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/lrn.cpp
index 6ab0256..ab4c2a9 100644 (file)
--- a/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/lrn.cpp
+++ b/inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/lrn.cpp
@@ -27,7 +27,7 @@ INSTANTIATE_TEST_CASE_P(LrnCheck, LrnLayerTest,
                                            ::testing::Values(beta),
                                            ::testing::Values(bias),
                                            ::testing::Values(size),
-                                           ::testing::Values(std::vector<size_t>({1})),
+                                           ::testing::Values(std::vector<int64_t>({1})),
                                            ::testing::ValuesIn(netPrecisions),
                                            ::testing::Values(std::vector<size_t>({10, 10, 3, 2})),
                                            ::testing::Values(CommonTestUtils::DEVICE_CPU)),

diff --git a/inference-engine/tests/functional/plugin/gpu/shared_tests_instances/single_layer_tests/lrn.cpp b/inference-engine/tests/functional/plugin/gpu/shared_tests_instances/single_layer_tests/lrn.cpp
index d49a323..92aeb59 100644 (file)
--- a/inference-engine/tests/functional/plugin/gpu/shared_tests_instances/single_layer_tests/lrn.cpp
+++ b/inference-engine/tests/functional/plugin/gpu/shared_tests_instances/single_layer_tests/lrn.cpp
@@ -25,7 +25,7 @@ INSTANTIATE_TEST_CASE_P(LrnCheck, LrnLayerTest,
                                            ::testing::Values(beta),
                                            ::testing::Values(bias),
                                            ::testing::Values(size),
-                                           ::testing::Values(std::vector<size_t>({1})),
+                                           ::testing::Values(std::vector<int64_t>({1})),
                                            ::testing::ValuesIn(netPrecisions),
                                            ::testing::Values(std::vector<size_t>({10, 10, 3, 2})),
                                            ::testing::Values(CommonTestUtils::DEVICE_GPU)),

diff --git a/inference-engine/tests/functional/plugin/shared/include/single_layer_tests/lrn.hpp b/inference-engine/tests/functional/plugin/shared/include/single_layer_tests/lrn.hpp
index 3494bad..4940c41 100644 (file)
--- a/inference-engine/tests/functional/plugin/shared/include/single_layer_tests/lrn.hpp
+++ b/inference-engine/tests/functional/plugin/shared/include/single_layer_tests/lrn.hpp
@@ -22,7 +22,7 @@ typedef std::tuple<
         double,                        // Beta
         double,                        // Bias
         size_t,                        // Size
-        std::vector<size_t>,           // Reduction axes
+        std::vector<int64_t>,          // Reduction axes
         InferenceEngine::Precision,    // Network precision
         InferenceEngine::SizeVector,   // Input shapes
         std::string                    // Device name

diff --git a/inference-engine/tests/functional/plugin/shared/src/single_layer_tests/lrn.cpp b/inference-engine/tests/functional/plugin/shared/src/single_layer_tests/lrn.cpp
index c2dc0a9..75e3a81 100644 (file)
--- a/inference-engine/tests/functional/plugin/shared/src/single_layer_tests/lrn.cpp
+++ b/inference-engine/tests/functional/plugin/shared/src/single_layer_tests/lrn.cpp
@@ -15,7 +15,7 @@ namespace LayerTestsDefinitions {
 std::string LrnLayerTest::getTestCaseName(testing::TestParamInfo<lrnLayerTestParamsSet> obj) {
     double alpha, beta, bias;
     size_t size;
-    std::vector<size_t> axes;
+    std::vector<int64_t> axes;
     InferenceEngine::Precision  netPrecision;
     std::vector<size_t> inputShapes;
     std::string targetDevice;
@@ -40,7 +40,7 @@ void LrnLayerTest::SetUp() {
     auto netPrecision   = InferenceEngine::Precision::UNSPECIFIED;
     double alpha, beta, bias;
     size_t size;
-    std::vector<size_t> axes;
+    std::vector<int64_t> axes;
     std::tie(alpha, beta, bias, size, axes, netPrecision, inputShapes, targetDevice) = GetParam();
 
     auto ngPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(netPrecision);