diff options
Diffstat (limited to 'model-optimizer/extensions/front/kaldi/eliminate_redundant_reshape.py')
-rw-r--r-- | model-optimizer/extensions/front/kaldi/eliminate_redundant_reshape.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/model-optimizer/extensions/front/kaldi/eliminate_redundant_reshape.py b/model-optimizer/extensions/front/kaldi/eliminate_redundant_reshape.py index 9050e8dfe..b76f5405c 100644 --- a/model-optimizer/extensions/front/kaldi/eliminate_redundant_reshape.py +++ b/model-optimizer/extensions/front/kaldi/eliminate_redundant_reshape.py @@ -36,9 +36,8 @@ class EliminateRedundantReshape(FrontReplacementPattern): nodes=[ ('reshape', dict(kind='op', op='Reshape')) ], - edges=[], - node_attrs=['kind', 'op'], - edge_attrs=[]) + edges=[] + ) @staticmethod def replace_pattern(graph: nx.MultiDiGraph, match: dict): |