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

Fix compilation error in template_plugin tests (#9248)

Browse Source
This commit is contained in:
Mateusz Tabaka authored and GitHub committed 2021-12-16 01:52:26 +03:00
1 parent d1e54d9961
commit 40f668140e
2 files changed
+8 -8
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

docs/template_plugin/tests/functional/op_reference/einsum.cpp
+4 -4
View File
@@ -30,7 +30,7 @@ class ReferenceEinsumTest : public testing::TestWithParam<EinsumParams>, public
public:
void SetUp() override {
auto params = GetParam();
function = CreateFunction(params);
function = CreateModel(params);
for (const auto& input_tensor : params.inputs) {
inputData.push_back(input_tensor.data);
}
@@ -52,7 +52,7 @@ public:
}
private:
static std::shared_ptr<Function> CreateFunction(const EinsumParams& params) {
static std::shared_ptr<Model> CreateModel(const EinsumParams& params) {
OutputVector output_vector;
ParameterVector param_vector;
for (const auto& input_tensor : params.inputs) {
@@ -61,7 +61,7 @@ private:
param_vector.push_back(param);
}
const auto einsum = std::make_shared<opset7::Einsum>(output_vector, params.equation);
const auto f = std::make_shared<Function>(OutputVector{einsum}, param_vector);
const auto f = std::make_shared<Model>(OutputVector{einsum}, param_vector);
return f;
}
};
@@ -179,4 +179,4 @@ std::vector<EinsumParams> generateCombinedParams() {
INSTANTIATE_TEST_SUITE_P(smoke_Einsum_With_Hardcoded_Refs, ReferenceEinsumTest,
testing::ValuesIn(generateCombinedParams()), ReferenceEinsumTest::getTestCaseName);
} // namespace
} // namespace
docs/template_plugin/tests/functional/op_reference/extract_image_patches.cpp
+4 -4
View File
@@ -36,7 +36,7 @@ class ReferenceExtractImagePatchesTest : public testing::TestWithParam<ExtractIm
public:
void SetUp() override {
auto params = GetParam();
function = CreateFunction(params);
function = CreateModel(params);
inputData = {params.data.data};
refOutData = {params.expectedResult.data};
}
@@ -59,14 +59,14 @@ public:
}
private:
static std::shared_ptr<Function> CreateFunction(const ExtractImagePatchesParams& params) {
static std::shared_ptr<Model> CreateModel(const ExtractImagePatchesParams& params) {
const auto data = std::make_shared<opset1::Parameter>(params.data.type, params.data.shape);
const auto extrace_image_patches = std::make_shared<opset3::ExtractImagePatches>(data,
params.sizes,
params.strides,
params.rates,
params.autoPad);
const auto f = std::make_shared<Function>(extrace_image_patches, ParameterVector{data});
const auto f = std::make_shared<Model>(extrace_image_patches, ParameterVector{data});
return f;
}
};
@@ -243,4 +243,4 @@ std::vector<ExtractImagePatchesParams> generateCombinedParams() {
INSTANTIATE_TEST_SUITE_P(smoke_ExtractImagePatches_With_Hardcoded_Refs, ReferenceExtractImagePatchesTest,
testing::ValuesIn(generateCombinedParams()), ReferenceExtractImagePatchesTest::getTestCaseName);
} // namespace
} // namespace