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

avoid can_fuse_reorder_to_prev() for detection_output cpu impl (#14350)

Browse Source
This commit is contained in:
Wilson Seok
2022-12-02 09:17:42 +09:00
committed by GitHub
parent 5e24219446
commit f7849ef465
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/plugins/intel_gpu/src/graph/layout_optimizer.cpp
View File

@@ -423,7 +423,8 @@ bool layout_optimizer::can_fuse_reorder_to_prev(program_node& prev, reorder_node
return false;
// Ref kernels are the main for depth_to_space, region_yolo and detection_output. It can do anything. Should not see next.
if (prev.is_type<depth_to_space>() || prev.is_type<region_yolo>() || prev.is_type<detection_output>())
if (prev.is_type<depth_to_space>() || prev.is_type<region_yolo>()
|| (prev.is_type<detection_output>() && prev.get_preferred_impl_type() != cldnn::impl_types::cpu))
return true;
if (node.get_users().empty())