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

[IE CLDNN] crop + NMS should not be merged because optimizing out crop will add unexpected padding while NMS cannot handle it properly (#6015)

Browse Source
This commit is contained in:
Taylor Yeonbok Lee 2021-06-04 18:19:40 +09:00 committed by GitHub
parent 90a93be071
commit d3a5c67f3b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
inference-engine/thirdparty/clDNN/src/graph_optimizer/prepare_buffer_fusing.cpp vendored
View File

@ -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"
@ -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;
}