Enable force shape inference in RNNSeqNormilize transformation (#2069)

diff --git a/model-optimizer/extensions/middle/RNNSequenceNormalizeToIE.py b/model-optimizer/extensions/middle/RNNSequenceNormalizeToIE.py
index e5de8c4..ccde2ef 100644 (file)
--- a/model-optimizer/extensions/middle/RNNSequenceNormalizeToIE.py
+++ b/model-optimizer/extensions/middle/RNNSequenceNormalizeToIE.py
@@ -51,6 +51,8 @@ class RNNSequenceNormalize(MiddleReplacementPattern):
             6: (optional for LSTM) Peepholes weights, shape  [(M - 1) * hidden_size]
 
     """
+    force_shape_inference = True
+
     def run_after(self):
         from extensions.middle.DecomposeBidirectionalRNNSequence import DecomposeBidirectionalRNNSequence
         return [DecomposeBidirectionalRNNSequence]