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

[IE CLDNN] Fix performance degradation for brain-tumor-segmentation-0002 (#2038)

Browse Source 
JIRA: 37933

Co-authored-by: Egor Churaev <egor.churaev@intel.com>
This commit is contained in:
Vladimir Paramuzov 2020-09-04 11:54:59 +03:00 committed by GitHub
parent 35357f686d
commit 214b25c49d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
inference-engine/thirdparty/clDNN/src/graph_optimizer/prepare_padding.cpp vendored
View File

@ -130,6 +130,15 @@ void prepare_padding::run(program_impl& p) {
continue; continue;
} }
// convolution have only one input primitive
auto prev_prim_output_layout = conv_input_node.get_output_layout();
// For 3d convolution padding is needed only for int8 case
// FP16/32 kernels can work w/o physical padding
if (prev_prim_output_layout.format == cldnn::format::b_fs_zyx_fsv16 &&
prev_prim_output_layout.data_type != data_types::i8 && prev_prim_output_layout.data_type != data_types::u8)
continue;
// We shoudn't apply any padding to nodes which are marked as outputs // We shoudn't apply any padding to nodes which are marked as outputs
if (conv_input_node.is_output()) if (conv_input_node.is_output())
continue; continue;
@ -140,9 +149,6 @@ void prepare_padding::run(program_impl& p) {
layout filter_layout = filter_node.get_output_layout(); layout filter_layout = filter_node.get_output_layout();
// convolution have only one input primitive
auto prev_prim_output_layout = conv_input_node.get_output_layout();
// Compute initial required paddings for primitive used as input for convolution. // Compute initial required paddings for primitive used as input for convolution.
auto input_offset = conv->input_offset; auto input_offset = conv->input_offset;
auto stride = conv->stride; auto stride = conv->stride;