experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a test case for a larger QR decomposition.

Browse Source 
PiperOrigin-RevId: 347464495
Change-Id: Id0b54d1cdfc88a544c1c698664580eebd2d9823e
This commit is contained in:
Peter Hawkins 2020-12-14 14:03:13 -08:00 committed by TensorFlower Gardener
parent c025ecf146
commit 3ae790096c
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/tests/qr_op_test.py
View File

@ -19,6 +19,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import itertools import itertools
import unittest
from absl.testing import parameterized from absl.testing import parameterized
import numpy as np import numpy as np
@ -129,6 +130,11 @@ class QrOpTest(xla_test.XLATestCase, parameterized.TestCase):
x_np = self._random_matrix(np.float32, (2000, 2000)) x_np = self._random_matrix(np.float32, (2000, 2000))
self._test(x_np, full_matrices=True) self._test(x_np, full_matrices=True)
@unittest.skip("Test times out on CI")
def testLarge17500x128(self):
x_np = self._random_matrix(np.float32, (17500, 128))
self._test(x_np, full_matrices=True)
@parameterized.parameters((23, 25), (513, 23)) @parameterized.parameters((23, 25), (513, 23))
def testZeroColumn(self, rows, cols): def testZeroColumn(self, rows, cols):
x_np = self._random_matrix(np.complex64, (rows, cols)) x_np = self._random_matrix(np.complex64, (rows, cols))