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

2021-11-17 19:44:19 +03:00 · 2021-11-17 19:44:19 +03:00 · 2fc4856b8a
commit 2fc4856b8a
parent 58b0793386
1 changed files with 1 additions and 1 deletions
--- 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;
            }
        }