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

[GPU] Matrix NMS (#13137)

Browse Source
This commit is contained in:
Konstantin Beluchenko authored and GitHub committed 2022-11-02 18:16:57 +04:00
1 parent f8b12b4f1a
commit c6528ee4ea
21 files changed
+1858 -3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/common/transformations/src/transformations/op_conversions/convert_matrix_nms_to_matrix_nms_ie.cpp
+3 -2
View File
@@ -14,6 +14,7 @@
#include "itt.hpp"
#include "ngraph_ops/nms_static_shape_ie.hpp"
#include "transformations/utils/utils.hpp"
ngraph::pass::ConvertMatrixNmsToMatrixNmsIE::ConvertMatrixNmsToMatrixNmsIE(bool force_i32_output_type) {
MATCHER_SCOPE(ConvertMatrixNmsToMatrixNmsIE);
@@ -46,13 +47,13 @@ ngraph::pass::ConvertMatrixNmsToMatrixNmsIE::ConvertMatrixNmsToMatrixNmsIE(bool
if (nms->output(1).get_element_type() != output_1.get_element_type()) {
output_1 = std::make_shared<opset1::Convert>(output_1, nms->output(1).get_element_type());
output_1.get_node_shared_ptr()->set_friendly_name(nms->get_friendly_name() + "/convert.1");
output_1.get_node_shared_ptr()->set_friendly_name(op::util::create_ie_output_name(nms->output(1)));
new_ops.emplace_back(output_1.get_node_shared_ptr());
}
if (nms->output(2).get_element_type() != output_2.get_element_type()) {
output_2 = std::make_shared<opset1::Convert>(output_2, nms->output(2).get_element_type());
output_2.get_node_shared_ptr()->set_friendly_name(nms->get_friendly_name() + "/convert.2");
output_2.get_node_shared_ptr()->set_friendly_name(op::util::create_ie_output_name(nms->output(2)));
new_ops.emplace_back(output_2.get_node_shared_ptr());
}