[CONFORMANCE] Fix for Eye-9 op in the Opset Conformance report (#19404)

This commit is contained in:
Irina Efode 2023-08-25 19:22:18 +04:00 committed by GitHub
parent f962511a84
commit cddcec8ba8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -379,7 +379,7 @@ std::shared_ptr<ov::Model> generate(const std::shared_ptr<ov::op::v3::ExtractIma
}
std::shared_ptr<ov::Model> generate(const std::shared_ptr<ov::op::v9::Eye> &node) {
const auto rows = ngraph::builder::makeConstant<int64_t>(ov::element::i64, {1}, {3});
const auto rows = std::make_shared<ov::op::v0::Parameter>(ov::element::i64, ov::Shape{1});
const auto cols = ngraph::builder::makeConstant<int64_t>(ov::element::i64, {1}, {4});
const auto diag = ngraph::builder::makeConstant<int64_t>(ov::element::i64, {1}, {0});
const auto batch = ngraph::builder::makeConstant<int64_t>(ov::element::i64, {3}, {2, 2, 2});
@ -388,9 +388,8 @@ std::shared_ptr<ov::Model> generate(const std::shared_ptr<ov::op::v9::Eye> &node
diag,
batch,
ov::element::f32);
ov::pass::disable_constant_folding(eye);
ov::ResultVector results{std::make_shared<ov::op::v0::Result>(eye)};
return std::make_shared<ov::Model>(results, ngraph::ParameterVector{}, "Eye");
return std::make_shared<ov::Model>(results, ngraph::ParameterVector{rows}, "Eye");
}
std::shared_ptr<ov::Model> generate(const std::shared_ptr<ov::op::v0::FakeQuantize> &node) {