Skip to content

Commit 60ea816

Browse files
[GPU] Add calculating max padding of reshape from input padding and current output padding
1 parent 7bcfaa5 commit 60ea816

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/plugins/intel_gpu/src/graph/reshape.cpp

+3-1
Original file line numberDiff line numberDiff line change
@@ -228,7 +228,9 @@ std::vector<layout> reshape_inst::calc_output_layouts(reshape_node const& node,
228228
output_format = node.get_preferred_output_fmt();
229229
}
230230

231-
return { layout {output_shapes[0], input_layout.data_type, format::adjust_to_rank(output_format, output_shapes[0].size()), out_pad} };
231+
auto max_out_pad = padding::max(impl_param.get_output_layout(0).data_padding, out_pad);
232+
233+
return { layout {output_shapes[0], input_layout.data_type, format::adjust_to_rank(output_format, output_shapes[0].size()), max_out_pad} };
232234
}
233235

234236
template std::vector<layout> reshape_inst::calc_output_layouts<ov::PartialShape>(reshape_node const& node, const kernel_impl_params& impl_param);

0 commit comments

Comments
 (0)