Merge pull request #43 from vlasenkoalexey/master

added support for rewriting function graphs in TF2.x models
CODAIT · Aug 6, 2021 · f3e0c85 · f3e0c85
2 parents c9e5a26 + a2993b5
commit f3e0c85
Show file tree

Hide file tree

Showing 13 changed files with 1,841 additions and 560 deletions.
diff --git a/graph_def_editor/__init__.py b/graph_def_editor/__init__.py
@@ -23,6 +23,7 @@
 from __future__ import print_function
 
 # pylint: disable=wildcard-import
+from graph_def_editor.base_graph import *
 from graph_def_editor.edit import *
 from graph_def_editor.graph import *
 from graph_def_editor.match import *