(self, rows, cols, dtype)
| 101 | |
| 102 | @parameterized.parameters(*PARAMS) |
| 103 | def testQR(self, rows, cols, dtype): |
| 104 | # TODO(b/111317468): Test other types. |
| 105 | for full_matrices in [True, False]: |
| 106 | # Only tests the (3, 2) case for small numbers of rows/columns. |
| 107 | for batch_dims in [(), (3,)] + [(3, 2)] * (max(rows, cols) < 10): |
| 108 | self._test(dtype, batch_dims + (rows, cols), full_matrices) |
| 109 | |
| 110 | |
| 111 | if __name__ == "__main__": |