IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE TESTS] Move ROI SLT class to the shared lib (#4198)

Browse Source 
* [IE TESTS] Move ROI SLT class to the shared lib

* [IE Tests] Fix Ci

* Fix ci
This commit is contained in:
Irina Efode
2021-02-08 13:56:34 +03:00
committed by GitHub
parent 0acd153a6c
commit bd967a1b91
3 changed files with 46 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
inference-engine/tests/functional/plugin/shared/include/single_layer_tests/roi_align.hpp
View File

@@ -5,42 +5,12 @@
#pragma once
#include <tuple>
#include <string>
#include <vector>
#include <memory>
#include "ngraph_functions/builders.hpp"
#include "ngraph_functions/utils/ngraph_helpers.hpp"
#include "shared_test_classes/base/layer_test_utils.hpp"
#include "shared_test_classes/single_layer/roi_align.hpp"
namespace LayerTestsDefinitions {
using roialignParams = std::tuple<std::vector<size_t>, // feature map shape
std::vector<size_t>, // proposal coords shape
int, // bin's row count
int, // bin's column count
float, // spatial scale
int, // pooling ratio
std::string, // pooling mode
InferenceEngine::Precision, // net precision
LayerTestsUtils::TargetDevice>; // device name
class ROIAlignLayerTest : public testing::WithParamInterface<roialignParams>,
virtual public LayerTestsUtils::LayerTestsCommon {
public:
static std::string getTestCaseName(testing::TestParamInfo<roialignParams> obj);
void Infer() override;
protected:
void SetUp() override;
private:
int pooledH;
int pooledW;
float spatialScale;
int poolingRatio;
std::string poolingMode;
};
TEST_P(ROIAlignLayerTest, CompareWithRefs) {
Run();
}
} // namespace LayerTestsDefinitions

39
inference-engine/tests/functional/shared_test_classes/include/shared_test_classes/single_layer/roi_align.hpp Normal file
View File

@@ -0,0 +1,39 @@
// Copyright (C) 2020 Intel Corporation
//
// SPDX-License-Identifier: Apache-2.0
//
#pragma once
#include "shared_test_classes/base/layer_test_utils.hpp"
namespace LayerTestsDefinitions {
using roialignParams = std::tuple<std::vector<size_t>, // feature map shape
std::vector<size_t>, // proposal coords shape
int, // bin's row count
int, // bin's column count
float, // spatial scale
int, // pooling ratio
std::string, // pooling mode
InferenceEngine::Precision, // net precision
LayerTestsUtils::TargetDevice>; // device name
class ROIAlignLayerTest : public testing::WithParamInterface<roialignParams>,
virtual public LayerTestsUtils::LayerTestsCommon {
public:
static std::string getTestCaseName(testing::TestParamInfo<roialignParams> obj);
void Infer() override;
protected:
void SetUp() override;
private:
int pooledH;
int pooledW;
float spatialScale;
int poolingRatio;
std::string poolingMode;
};
} // namespace LayerTestsDefinitions

15
inference-engine/tests/functional/plugin/shared/src/single_layer_tests/roi_align.cpp → inference-engine/tests/functional/shared_test_classes/src/single_layer/roi_align.cpp
View File

@@ -3,15 +3,10 @@
// SPDX-License-Identifier: Apache-2.0
//
#include <tuple>
#include <string>
#include <vector>
#include <memory>
#include <ngraph/opsets/opset3.hpp>
#include "common_test_utils/common_utils.hpp"
#include "functional_test_utils/skip_tests_config.hpp"
#include "shared_test_classes/base/layer_test_utils.hpp"
#include "single_layer_tests/roi_align.hpp"
#include "ngraph_functions/builders.hpp"
#include "shared_test_classes/single_layer/roi_align.hpp"
using namespace InferenceEngine;
using namespace FuncTestUtils::PrecisionUtils;
@@ -129,8 +124,4 @@ void ROIAlignLayerTest::SetUp() {
ngraph::ResultVector results{std::make_shared<ngraph::opset3::Result>(roiAlign)};
function = std::make_shared<ngraph::Function>(results, params, "roi_align");
}
TEST_P(ROIAlignLayerTest, CompareWithRefs) {
Run();
}
} // namespace LayerTestsDefinitions