diff options
| author |  Yangzihao Wang <yangzihao@google.com> | 2017-07-27 11:08:47 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-27 11:12:54 -0700 |
| commit | 3b97f1f9bd983dc4ccc8234ee297cf69d397f0c8 (patch) | |
| tree | 9aeb69897bcdec11e9a74760ad5e091f3e71f8c5 /tensorflow | |
| parent | ae3119d16ba475f25c46f5f21d617d232c904674 (diff) | |
Change to only run one round of matmul benchmark.
PiperOrigin-RevId: 163364341
Diffstat (limited to 'tensorflow')
| -rw-r--r-- | tensorflow/python/ops/matmul_benchmark.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/python/ops/matmul_benchmark.py b/tensorflow/python/ops/matmul_benchmark.py index 55c575162a..b777ace9d0 100644 --- a/tensorflow/python/ops/matmul_benchmark.py +++ b/tensorflow/python/ops/matmul_benchmark.py @@ -106,7 +106,7 @@ class MatmulBenchmark(test.Benchmark): device=device, dtype=str(dtype).replace(' ', ''), inputinfo=str(n) + 'x' + str(m) + 'x' + str(k) + ',ta:' + - str(transpose_a) + '.tb:' + str(transpose_b)).replace(' ', ''), + str(transpose_a) + ',tb:' + str(transpose_b)).replace(' ', ''), iters=num_iters, wall_time=duration) return duration @@ -134,9 +134,7 @@ class MatmulBenchmark(test.Benchmark): self.run_test_gpu(n, m, k, transpose_a, transpose_b, dtype, num_iters) def benchmark_matmul(self): - num_iters = 200 - for _ in range(10): - self.test_round(num_iters) + self.test_round(num_iters=200) if __name__ == '__main__': |