Skip to content

Commit 08a7a71

Browse files
author
Liubov Batanina
committed
Fixed type of reduction axes
1 parent 0509c66 commit 08a7a71

File tree

4 files changed

+5
-5
lines changed
  • inference-engine/tests/functional/plugin
    • cpu/shared_tests_instances/single_layer_tests
    • gpu/shared_tests_instances/single_layer_tests
    • shared

4 files changed

+5
-5
lines changed

inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/lrn.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ INSTANTIATE_TEST_CASE_P(LrnCheck, LrnLayerTest,
2727
::testing::Values(beta),
2828
::testing::Values(bias),
2929
::testing::Values(size),
30-
::testing::Values(std::vector<size_t>({1})),
30+
::testing::Values(std::vector<int64_t>({1})),
3131
::testing::ValuesIn(netPrecisions),
3232
::testing::Values(std::vector<size_t>({10, 10, 3, 2})),
3333
::testing::Values(CommonTestUtils::DEVICE_CPU)),

inference-engine/tests/functional/plugin/gpu/shared_tests_instances/single_layer_tests/lrn.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ INSTANTIATE_TEST_CASE_P(LrnCheck, LrnLayerTest,
2525
::testing::Values(beta),
2626
::testing::Values(bias),
2727
::testing::Values(size),
28-
::testing::Values(std::vector<size_t>({1})),
28+
::testing::Values(std::vector<int64_t>({1})),
2929
::testing::ValuesIn(netPrecisions),
3030
::testing::Values(std::vector<size_t>({10, 10, 3, 2})),
3131
::testing::Values(CommonTestUtils::DEVICE_GPU)),

inference-engine/tests/functional/plugin/shared/include/single_layer_tests/lrn.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ typedef std::tuple<
2222
double, // Beta
2323
double, // Bias
2424
size_t, // Size
25-
std::vector<size_t>, // Reduction axes
25+
std::vector<int64_t>, // Reduction axes
2626
InferenceEngine::Precision, // Network precision
2727
InferenceEngine::SizeVector, // Input shapes
2828
std::string // Device name

inference-engine/tests/functional/plugin/shared/src/single_layer_tests/lrn.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ namespace LayerTestsDefinitions {
1515
std::string LrnLayerTest::getTestCaseName(testing::TestParamInfo<lrnLayerTestParamsSet> obj) {
1616
double alpha, beta, bias;
1717
size_t size;
18-
std::vector<size_t> axes;
18+
std::vector<int64_t> axes;
1919
InferenceEngine::Precision netPrecision;
2020
std::vector<size_t> inputShapes;
2121
std::string targetDevice;
@@ -40,7 +40,7 @@ void LrnLayerTest::SetUp() {
4040
auto netPrecision = InferenceEngine::Precision::UNSPECIFIED;
4141
double alpha, beta, bias;
4242
size_t size;
43-
std::vector<size_t> axes;
43+
std::vector<int64_t> axes;
4444
std::tie(alpha, beta, bias, size, axes, netPrecision, inputShapes, targetDevice) = GetParam();
4545

4646
auto ngPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(netPrecision);

0 commit comments

Comments
 (0)