From 2d606e35c57757b6de090df904654e0ee114f4c9 Mon Sep 17 00:00:00 2001
From: Ivan Tikhonov <ivan.tikhonov@intel.com>
Date: Fri, 4 Sep 2020 15:08:52 +0300
Subject: [PATCH] Enable force shape inference in RNNSeqNormilize
 transformation (#2069)

---
 model-optimizer/extensions/middle/RNNSequenceNormalizeToIE.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/model-optimizer/extensions/middle/RNNSequenceNormalizeToIE.py b/model-optimizer/extensions/middle/RNNSequenceNormalizeToIE.py
index e5de8c47b35..ccde2efcbdd 100644
--- 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]