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

Update Model::reshape to rethrow currently handled exception (#9723)

Browse Source 
* Update Model::reshape to rethrow the currently handled exception

* Set expected type of exception in TestInvalidReshape
This commit is contained in:
Katarzyna Mitrus 2022-01-18 09:49:02 +01:00 committed by GitHub
parent 89f4a569d5
commit 18309852ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/core/src/model.cpp
View File

@ -877,10 +877,10 @@ void ov::Model::reshape(const std::map<ov::Output<ov::Node>, ov::PartialShape>&
ssr_manager.run_passes(shared_from_this());
reshape_only(new_param_shapes);
} catch (std::exception& ex) {
} catch (...) {
// restore shapes to original ones
reshape_only(original_input_shapes);
throw ex;
throw;
}
}

2
src/core/tests/model.cpp
View File

@ -741,7 +741,7 @@ TEST(model_reshape, TestInvalidReshape) {
f = std::make_shared<ov::Model>(ov::OutputVector{reshape}, ov::ParameterVector{input});
}
EXPECT_ANY_THROW(f->reshape({{"tensor", ov::Shape({4})}}));
EXPECT_THROW(f->reshape({{"tensor", ov::Shape({4})}}), ov::Exception);
auto param = f->get_parameters().front();
EXPECT_EQ(param->get_output_shape(0), ov::Shape({1, 1000, 4}));