diff options
Diffstat (limited to 'model-optimizer/extensions/front/kaldi/replace_lstm_node_pattern.py')
-rw-r--r-- | model-optimizer/extensions/front/kaldi/replace_lstm_node_pattern.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/model-optimizer/extensions/front/kaldi/replace_lstm_node_pattern.py b/model-optimizer/extensions/front/kaldi/replace_lstm_node_pattern.py index 856867103..e9bd31fc3 100644 --- a/model-optimizer/extensions/front/kaldi/replace_lstm_node_pattern.py +++ b/model-optimizer/extensions/front/kaldi/replace_lstm_node_pattern.py @@ -81,6 +81,7 @@ def create_node(graph: nx.MultiDiGraph, name: str, attrs: dict, inputs: tuple = attrs['fw_tensor_debug_info'] = [(Node(graph, noe_id).soft_get('name'), None)] if index < len(out_indexes): attrs['out'] = out_indexes[index] + attrs['in'] = index edges.append((noe_id, new_graph_node.id, attrs)) graph.add_edges_from(edges) |