Update TensorFlow script

lutzroeder · Apr 30, 2024 · 32228b8 · 32228b8
1 parent f406e54
commit 32228b8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/tf_metadata.py b/tools/tf_metadata.py
@@ -3,6 +3,7 @@
 import json
 import logging
 import os
+import re
 import google.protobuf # pylint: disable=import-error
 
 logging.getLogger('tensorflow').setLevel(logging.ERROR)
@@ -87,6 +88,7 @@ def _pbtxt_from_multiline(multiline_pbtxt):
 def _read_op_list(file):
     op_list = op_def_pb2.OpList() # pylint: disable=no-member
     content = _read(file)
+    content = re.sub(r'^go/[a-z]+\s*', '', content)
     google.protobuf.text_format.Merge(content, op_list)
     return op_list