fix after code review

openvinotoolkit · Jul 29, 2020 · 8170cec · 8170cec
1 parent b26330b
commit 8170cec
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 12 deletions.
diff --git a/ngraph/src/ngraph/op/binary_convolution.cpp b/ngraph/src/ngraph/op/binary_convolution.cpp
@@ -87,10 +87,11 @@ void op::v1::BinaryConvolution::validate_and_infer_types()
         {
             result_shape[0] = data_batch_shape[0]; // batch size
         }
-    }
-    if (filters_shape.rank().is_static() && filters_shape.rank().get_length() > 1)
-    {
-        result_shape[1] = filters_shape[0]; // filter channel size
+
+        if (filters_shape.rank().is_static() && filters_shape.rank().get_length() > 1)
+        {
+            result_shape[1] = filters_shape[0]; // filter channel size
+        }
     }
 
     if (m_strides.size() == 0)

diff --git a/ngraph/src/ngraph/op/convolution.cpp b/ngraph/src/ngraph/op/convolution.cpp
@@ -72,10 +72,11 @@ void op::v1::Convolution::validate_and_infer_types()
         {
             result_shape[0] = data_batch_shape[0]; // batch size
         }
-    }
-    if (filters_shape.rank().is_static() && filters_shape.rank().get_length() > 1)
-    {
-        result_shape[1] = filters_shape[0]; // filter channel size
+
+        if (filters_shape.rank().is_static() && filters_shape.rank().get_length() > 1)
+        {
+            result_shape[1] = filters_shape[0]; // filter channel size
+        }
     }
 
     element::Type result_et;

diff --git a/ngraph/src/ngraph/op/deformable_convolution.cpp b/ngraph/src/ngraph/op/deformable_convolution.cpp
@@ -132,10 +132,11 @@ void op::v1::DeformableConvolution::validate_and_infer_types()
         {
             result_shape[0] = data_batch_shape[0]; // batch size
         }
-    }
-    if (filters_shape.rank().is_static() && filters_shape.rank().get_length() > 1)
-    {
-        result_shape[1] = filters_shape[0]; // filter channel size
+
+        if (filters_shape.rank().is_static() && filters_shape.rank().get_length() > 1)
+        {
+            result_shape[1] = filters_shape[0]; // filter channel size
+        }
     }
 
     if (m_strides.size() == 0)