diff options
author | Vijay Vasudevan <vrv@google.com> | 2017-09-06 16:14:04 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-06 16:18:01 -0700 |
commit | 903e2af10ab3012d853241504f51485c83267d43 (patch) | |
tree | 139ca6e1cc69598580e4d131c10fb8f6c6a86843 /tensorflow/contrib/tensorboard | |
parent | 713e36cedc4cefc9e6d33ed0a9aa76bfd8c78783 (diff) |
Change trace.py to not access a graph as a default argument.
Checks for None and access via default graph inside the function.
PiperOrigin-RevId: 167788815
Diffstat (limited to 'tensorflow/contrib/tensorboard')
-rw-r--r-- | tensorflow/contrib/tensorboard/plugins/trace/trace.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorboard/plugins/trace/trace.py b/tensorflow/contrib/tensorboard/plugins/trace/trace.py index 57f95dfce7..07e5316b8b 100644 --- a/tensorflow/contrib/tensorboard/plugins/trace/trace.py +++ b/tensorflow/contrib/tensorboard/plugins/trace/trace.py @@ -38,7 +38,7 @@ TOKENS = LEFT_TOKENS + RIGHT_TOKENS def store_trace_info(output_file_path, - graph=ops.get_default_graph(), + graph=None, ignore_regex_fpaths=None): """Collects and stores trace information for a TensorFlow model. @@ -51,6 +51,8 @@ def store_trace_info(output_file_path, in this list will be ignored. Defaults to patterns that match the core tensorflow python library. """ + graph = graph or ops.get_default_graph() + if not ignore_regex_fpaths: ignore_regex_fpaths = TF_LIB_REGEX_FPATHS |