diff options
-rw-r--r-- | tensorflow/contrib/linalg/python/kernel_tests/linear_operator_udvh_update_test.py | 9 |
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 |