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

[ v4::NMS ] Fixed v4 NMS cloning (#1113)

Browse Source
This commit is contained in:
Evgenya Stepyreva 2020-06-24 23:09:48 +03:00 committed by GitHub
parent 377531002c
commit f81257c969
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ngraph/src/ngraph/op/non_max_suppression.cpp
View File

@ -481,7 +481,7 @@ shared_ptr<Node>
? new_args.at(4)
: ngraph::op::Constant::create(element::f32, Shape{}, {.0f});
return std::make_shared<op::v3::NonMaxSuppression>(new_args.at(0),
return std::make_shared<op::v4::NonMaxSuppression>(new_args.at(0),
new_args.at(1),
arg2,
arg3,