diff options
Diffstat (limited to 'tensorflow/python/ops/batch_norm_benchmark.py')
-rw-r--r-- | tensorflow/python/ops/batch_norm_benchmark.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/ops/batch_norm_benchmark.py b/tensorflow/python/ops/batch_norm_benchmark.py index 397ed91078..c2ee2b3832 100644 --- a/tensorflow/python/ops/batch_norm_benchmark.py +++ b/tensorflow/python/ops/batch_norm_benchmark.py @@ -198,7 +198,7 @@ class BatchNormBenchmark(test.Benchmark): if FLAGS.use_gpu: t1 = self._run_graph("gpu", shape, axes, 10, "op", True, True, 50) t2 = self._run_graph("gpu", shape, axes, 10, "py", True, True, 50) - t2 = self._run_graph("gpu", shape, axes, 10, "slow", True, True, 50) + t3 = self._run_graph("gpu", shape, axes, 10, "slow", True, True, 50) print_difference("op vs py", t1, t2) print_difference("py vs slow", t2, t3) print("Forward convolution (higher layers).") |