Skip to content

Commit

Permalink
[IE CLDNN] crop + NMS should not be merged because optimizing out cro…
Browse files Browse the repository at this point in the history
…p will add unexpected padding while NMS cannot handle it properly (openvinotoolkit#6015)
  • Loading branch information
yeonbok authored Jun 4, 2021
1 parent 90a93be commit d3a5c67
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
#include "depth_to_space_inst.h"
#include "resample_inst.h"
#include "loop_inst.h"
#include "non_max_suppression_inst.h"

#include "pass_manager.h"
#include "program_helpers.h"
Expand Down Expand Up @@ -278,7 +279,7 @@ void prepare_buffer_fusing::run(program_impl& p) {
for (auto user : node.get_users()) {
if (user->is_type<concatenation>() && !user->is_output())
return;
if (user->is_type<loop>())
if (user->is_type<loop>() || user->is_type<non_max_suppression>())
return;
}

Expand Down

0 comments on commit d3a5c67

Please sign in to comment.