diff options
Diffstat (limited to 'tensorflow/python/client/timeline_test.py')
-rw-r--r-- | tensorflow/python/client/timeline_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/client/timeline_test.py b/tensorflow/python/client/timeline_test.py index e8797712e9..8396df5f40 100644 --- a/tensorflow/python/client/timeline_test.py +++ b/tensorflow/python/client/timeline_test.py @@ -100,8 +100,8 @@ class TimelineTest(test.TestCase): self.assertTrue(run_metadata.HasField('step_stats')) step_stats = run_metadata.step_stats devices = [d.device for d in step_stats.dev_stats] - self.assertTrue('/job:localhost/replica:0/task:0/gpu:0' in devices) - self.assertTrue('/gpu:0/stream:all' in devices) + self.assertTrue('/job:localhost/replica:0/task:0/device:GPU:0' in devices) + self.assertTrue('/device:GPU:0/stream:all' in devices) tl = timeline.Timeline(step_stats) ctf = tl.generate_chrome_trace_format() self._validateTrace(ctf) |