Enable force shape inference in RNNSeqNormilize transformation (#2069)

2020-09-04 15:08:52 +03:00 · 2020-09-04 15:08:52 +03:00 · 2d606e35c5
commit 2d606e35c5
parent 7212ff5bb2
1 changed files with 2 additions and 0 deletions
--- 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]