experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

substitute regex to match multi-line timeline_label

This commit is contained in:
zilinzhu 2020-04-24 17:24:58 +08:00
parent aee0c97173
commit ee13794ff4
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/client/timeline.py
View File

@ -400,9 +400,10 @@ class Timeline(object):
def _parse_kernel_label(self, label, node_name): def _parse_kernel_label(self, label, node_name):
"""Parses the fields in a node timeline label.""" """Parses the fields in a node timeline label."""
# Expects labels of the form: retval (arg) detail @@annotation # Expects labels of the form: retval (arg) detail @@annotation
match = re.match(r'.*@@(.*)\#id.*', label) start = label.find("@@")
if match is not None: end = label.find("#")
node_name = match.group(1) if start >= 0 and end >= 0 and start + 2 < end:
node_name = label[start+2:end]
# Node names should always have the form 'name:op'. # Node names should always have the form 'name:op'.
fields = node_name.split(':') + ['unknown'] fields = node_name.split(':') + ['unknown']
name, op = fields[:2] name, op = fields[:2]