[LPT] GroupConvolution: 3D tensor support (openvinotoolkit#8401)

openvino-dev-samples · Nov 23, 2021 · 6ae4833 · 6ae4833
1 parent fd40215
commit 6ae4833
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference-engine/src/low_precision_transformations/src/weightable_layer_transformation.cpp b/inference-engine/src/low_precision_transformations/src/weightable_layer_transformation.cpp
@@ -96,7 +96,7 @@ bool WeightableLayerTransformation::canBeTransformed(const TransformationContext
             }
 
             const auto rankVal = rank.get_length();
-            if ((rankVal != 4) && (rankVal != 5)) {
+            if ((rankVal < 3) && (rankVal > 5)) {
                 return false;
             }
         }