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

add GenerateProposals single layer test (#12933)

Browse Source 
Co-authored-by: Andrei Kochin <andrei.kochin@intel.com>
This commit is contained in:
mei, yang 2022-09-08 22:35:14 +08:00 committed by GitHub
parent 7b98ebf627
commit b2d37ff85a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/tests/functional/plugin/shared/src/single_layer_tests/op_impl_check/single_op_graph.cpp
View File

@ -1660,6 +1660,25 @@ std::shared_ptr<ov::Model> generateGatherNDBase(const std::shared_ptr<ov::op::Op
return std::make_shared<ov::Model>(results, params, "GatherNDBaseGraph");
}
std::shared_ptr<ov::Model> generate(const std::shared_ptr<ov::op::v9::GenerateProposals> &node) {
const auto params = ngraph::builder::makeDynamicParams(ov::element::f32, {{1, 3}, {2, 2, 3, 4}, {1, 12, 2, 2}, {1, 3, 2, 2}});
const auto outputs =
ngraph::helpers::convert2OutputVector(ngraph::helpers::castOps2Nodes<ov::op::v0::Parameter>(params));
ov::op::v9::GenerateProposals::Attributes attrs;
attrs.min_size = 1;
attrs.nms_threshold = 0.8;
attrs.pre_nms_count = 100;
attrs.post_nms_count = 100;
if (ov::is_type<ov::op::v9::GenerateProposals>(node)) {
const auto gp = std::make_shared<ov::op::v9::GenerateProposals>(
outputs[0], outputs[1], outputs[2], outputs[3], attrs);
ov::ResultVector results{std::make_shared<ov::op::v0::Result>(gp)};
return std::make_shared<ov::Model>(results, params, "GenerateProposalsGraph");
} else {
return nullptr;
}
}
std::shared_ptr<ov::Model> generateRNNCellBase(const std::shared_ptr<ov::op::Op> &node) {
std::shared_ptr<ov::Node> RNNCellBaseNode;
if (ov::is_type<ov::op::v3::GRUCell>(node)) {