diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-04 15:27:44 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-04 15:32:05 -0700 |
commit | 3a457c7252f09afd03483092ce9dcc7aa292b8c6 (patch) | |
tree | a58f0099aa9fb367e74a860b70985b7c276bec92 /tensorflow/python | |
parent | feda8c786948b1c7cc6bd9fe447781ceaff6b3d3 (diff) |
This CL fixes a bug in the eager benchmarks test that caused the defun tests to execute a different-sized matrix multiply than the eager tests.
PiperOrigin-RevId: 215814346
Diffstat (limited to 'tensorflow/python')
-rw-r--r-- | tensorflow/python/eager/benchmarks_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/eager/benchmarks_test.py b/tensorflow/python/eager/benchmarks_test.py index 3fe79ef244..2b0118c07f 100644 --- a/tensorflow/python/eager/benchmarks_test.py +++ b/tensorflow/python/eager/benchmarks_test.py @@ -353,7 +353,7 @@ class MicroBenchmarks(test.Benchmark): num_iters, execution_mode=None): f = function.defun(math_ops.matmul) - func = lambda: f(m, m, transpose_b) + func = lambda: f(m, m, transpose_b=transpose_b) self._run(func, num_iters, execution_mode=execution_mode) def _benchmark_defun_matmul_forward_backward(self, @@ -366,7 +366,7 @@ class MicroBenchmarks(test.Benchmark): def func(): with backprop.GradientTape() as gt: gt.watch(m) - y = f(m, m, transpose_b) + y = f(m, m, transpose_b=transpose_b) _ = gt.gradient(y, m) self._run(func, num_iters, execution_mode=execution_mode) |