diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/argmax_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/argmax_op_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/argmax_op_test.py b/tensorflow/python/kernel_tests/argmax_op_test.py index c24078a35a..1ad3dba2ae 100644 --- a/tensorflow/python/kernel_tests/argmax_op_test.py +++ b/tensorflow/python/kernel_tests/argmax_op_test.py @@ -54,7 +54,7 @@ class ArgMaxTest(tf.test.TestCase): x = np.asarray(100*np.random.randn(3, 2, 4, 5, 6), dtype=dtype) # Check that argmin and argmax match numpy along all dimensions - for dim in range(5): + for dim in range(-5, 5): self._testBothArg(tf.argmax, x, dim, x.argmax(dim)) self._testBothArg(tf.argmin, x, dim, x.argmin(dim)) |