fix bug

2020-02-27 12:08:59 +08:00 · 2020-02-27 12:08:59 +08:00 · 6ea53a1cfe
commit 6ea53a1cfe
parent 6c372f244b
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/python/client/timeline.py
+++ b/tensorflow/python/client/timeline.py
@ -620,6 +620,7 @@ class Timeline(object):
  def _preprocess_op_time(self, op_time):
    """Modified the start and end of ops in step stats"""
    if op_time == "schedule":
+      self._step_stats = self._origin_step_stats
      return
    self._step_stats = copy.deepcopy(self._origin_step_stats)
    # Separate job task and gpu tracer stream