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

Corrected Convolution checks

Browse Source
This commit is contained in:
Mikhail Ryzhov 2023-03-21 12:06:01 +01:00
parent 83f4428f48
commit e6650cfd1c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/plugins/intel_gna/src/backend/gna_limitations.cpp
View File

@ -99,8 +99,9 @@ bool is_conv_supported(const std::shared_ptr<ov::intel_gna::op::GNAConvolution>&
};
auto input_shape = conv_gna->input_value(0).get_shape();
auto filter_shape = conv_gna->input_value(1).get_shape();
if ((4 == filter_shape.size() && filter_shape[2] > 1 && filter_shape[3] > 1) ||
(4 == input_shape.size() && input_shape[2] > 1 && input_shape[3] > 1)) {
// GNAConvolution with NHWC layout
if ((4 == filter_shape.size() && filter_shape[1] > 1 && filter_shape[2] > 1) ||
(4 == input_shape.size() && input_shape[1] > 1 && input_shape[2] > 1)) {
pass::helper::ConvData conv_data;
ov::intel_gna::pass::helper::GetConvData(conv_gna, conv_data);
if (gna_convolution_layer::isMappableFrom2DTo1D(conv_data.input_height,