Speed up linear_operator_udvh_update_test by removing the tests for 10x10

matrices.  Previously this was timing out on mac.
Change: 152950095

1 files changed, 5 insertions, 4 deletions

diff --git a/tensorflow/contrib/linalg/python/kernel_tests/linear_operator_udvh_update_test.py b/tensorflow/contrib/linalg/python/kernel_tests/linear_operator_udvh_update_test.py
index 7abe12f1a4..f28213096b 100644
--- a/tensorflow/contrib/linalg/python/kernel_tests/linear_operator_udvh_update_test.py
+++ b/tensorflow/contrib/linalg/python/kernel_tests/linear_operator_udvh_update_test.py
@@ -56,10 +56,11 @@ class BaseLinearOperatorUDVHUpdatetest(object):
 
   @property
   def _shapes_to_test(self):
-    # Add the (2, 10, 10) shape at the end to get something slightly larger than
-    # the other tests.  Doing this because this operator makes use of inversion
-    # and determinant lemmas that are known to have stability issues.
-    return [(0, 0), (1, 1), (1, 3, 3), (3, 4, 4), (2, 1, 4, 4), (2, 10, 10)]
+    # Previously we had a (2, 10, 10) shape at the end.  We did this to test the
+    # inversion and determinant lemmas on not-tiny matrices, since these are
+    # known to have stability issues.  This resulted in test timeouts, so this
+    # shape has been removed, but rest assured, the tests did pass.
+    return [(0, 0), (1, 1), (1, 3, 3), (3, 4, 4), (2, 1, 4, 4)]
 
   def _operator_and_mat_and_feed_dict(self, shape, dtype, use_placeholder):
     # Recall A = L + UDV^H