Skip to content

Commit

Permalink
code style
Browse files Browse the repository at this point in the history
  • Loading branch information
CuriousPanCake committed Nov 12, 2024
1 parent 7b99815 commit bee461f
Showing 1 changed file with 6 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -109,16 +109,18 @@ ov::pass::ConvertNMS9ToNMSIEInternal::ConvertNMS9ToNMSIEInternal() {
Output<Node> output_0 = nms_legacy->output(0);
if (nms_9->output(0).get_element_type() != output_0.get_element_type()) {
output_0 = std::make_shared<ov::op::v0::Convert>(output_0, nms_9->output(0).get_element_type());
output_0.get_node_shared_ptr()->set_friendly_name(nms_9->output(0).get_node_shared_ptr()->get_friendly_name()
+ "." + std::to_string(output_0.get_index()));
output_0.get_node_shared_ptr()->set_friendly_name(
nms_9->output(0).get_node_shared_ptr()->get_friendly_name() + "." +
std::to_string(output_0.get_index()));
new_ops.emplace_back(output_0.get_node_shared_ptr());
}

Output<Node> output_2 = nms_legacy->output(2);
if (nms_9->output(2).get_element_type() != output_2.get_element_type()) {
output_2 = std::make_shared<ov::op::v0::Convert>(output_2, nms_9->output(2).get_element_type());
output_2.get_node_shared_ptr()->set_friendly_name(nms_9->output(2).get_node_shared_ptr()->get_friendly_name()
+ "." + std::to_string(output_2.get_index()));
output_2.get_node_shared_ptr()->set_friendly_name(
nms_9->output(2).get_node_shared_ptr()->get_friendly_name() + "." +
std::to_string(output_2.get_index()));
new_ops.emplace_back(output_2.get_node_shared_ptr());
}

Expand Down

0 comments on commit bee461f

Please sign in to comment.