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

Unbreak ShapeOf conformance test with dynamic shapes (#13326)

Browse Source
This commit is contained in:
avoskoboinyk-lohika 2022-10-17 21:47:14 +03:00 committed by GitHub
parent 5519fff5c2
commit daa8aaa894
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/tests/functional/plugin/conformance/test_runner/conformance_infra/src/read_ir_test/read_ir.cpp
View File

@ -93,6 +93,10 @@ void ReadIRTest::query_model() {
}
}
uint64_t clip(uint64_t n, uint64_t lower, uint64_t upper) {
return std::max(lower, std::min(n, upper));
}
void ReadIRTest::SetUp() {
// in case of crash jump will be made and work will be continued
auto crashHandler = std::unique_ptr<CommonTestUtils::CrashHandler>(new CommonTestUtils::CrashHandler());
@ -208,13 +212,14 @@ void ReadIRTest::SetUp() {
staticShapes[1] = midShape;
// Shape validation to avoid large values
for (auto& shape : staticShapes) {
for (auto& dim : shape) {
if (dim == 0) {
dim = 1;
} else if (dim > std::numeric_limits<char>::max()) {
dim = std::numeric_limits<char>::max();
}
uint64_t dimMin = 1;
uint64_t dimMax = std::numeric_limits<char>::max();
for (int i = 0; i < staticShapes[0].size(); ++i) {
auto& dim0 = staticShapes[0][i];
auto& dim2 = staticShapes[2][i];
if (dim0 != dim2) {
dim0 = clip(dim0, dimMin, dimMax);
dim2 = clip(dim2, dimMin, dimMax);
}
}
inputShapes.push_back(InputShape{param->get_partial_shape(), staticShapes});