when seq_len is given, identity_consumers will contain GRUBlockCell and Select. Sometime disorder of these consumers will lead to the failing of GRUBlockCell rewrite

Author: wayuanho

tf2onnx/rewriter/grublock_rewriter.py

@@ -90,7 +90,7 @@ def find_inputs(self, rnn_scope_name, rnn_props, match, input_blacklist=None):
     @staticmethod
     def _state_switch_check(enter_target_node_input_id, identity_consumers, match):
         node = match.get_op("GRUBlockCell")
-        if node == identity_consumers[0]:
+        if node in identity_consumers:
             log.debug("find state initializer value at %s", enter_target_node_input_id)
             return enter_target_node_input_id
         return None