• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Tests of TensorFlow linalg kernels written using the Python API.
2
3load("//tensorflow:tensorflow.bzl", "cuda_py_test")
4
5# buildifier: disable=same-origin-load
6load("//tensorflow:tensorflow.bzl", "tf_py_test")
7
8package(
9    default_visibility = ["//tensorflow:internal"],
10    licenses = ["notice"],
11)
12
13cuda_py_test(
14    name = "cholesky_op_test",
15    size = "medium",
16    srcs = ["cholesky_op_test.py"],
17    shard_count = 5,
18    tags = [
19        "nomsan",  # TODO(b/131773093): Re-enable.
20    ],
21    deps = [
22        "//tensorflow/python:array_ops",
23        "//tensorflow/python:client_testlib",
24        "//tensorflow/python:framework_for_generated_wrappers",
25        "//tensorflow/python:linalg_ops",
26        "//tensorflow/python:math_ops",
27        "//tensorflow/python:platform",
28        "//tensorflow/python/ops/linalg",
29        "//third_party/py/numpy",
30    ],
31)
32
33cuda_py_test(
34    name = "determinant_op_test",
35    size = "medium",
36    srcs = ["determinant_op_test.py"],
37    deps = [
38        "//tensorflow/python:client_testlib",
39        "//tensorflow/python:framework_for_generated_wrappers",
40        "//tensorflow/python:linalg_ops",
41        "//third_party/py/numpy",
42    ],
43)
44
45tf_py_test(
46    name = "eig_op_test",
47    size = "medium",
48    srcs = ["eig_op_test.py"],
49    data = ["//tensorflow/python/kernel_tests/linalg/testdata:self_adjoint_eig_op_test_files"],
50    shard_count = 20,
51    tags = [
52        "no_windows",
53        "nomsan",  # TODO(b/213581489): MSAN false positives from LAPACK use inside NumPy.
54    ],
55    deps = [
56        "//tensorflow/python:array_ops",
57        "//tensorflow/python:client_testlib",
58        "//tensorflow/python:framework_for_generated_wrappers",
59        "//tensorflow/python:linalg_ops",
60        "//tensorflow/python:math_ops",
61        "//third_party/py/numpy",
62    ],
63    # b/127344411: xla_enable_strict_auto_jit = True,
64)
65
66cuda_py_test(
67    name = "einsum_op_test",
68    size = "medium",
69    srcs = ["einsum_op_test.py"],
70    shard_count = 4,
71    xla_tags = [
72        "no_cuda_asan",  # times out
73    ],
74    deps = [
75        "//tensorflow/python:array_ops",
76        "//tensorflow/python:client_testlib",
77        "//tensorflow/python:framework_for_generated_wrappers",
78        "//tensorflow/python:linalg_ops",
79        "//tensorflow/python:math_ops",
80        "//tensorflow/python:platform",
81        "//tensorflow/python/ops/linalg",
82        "//third_party/py/numpy",
83    ],
84)
85
86cuda_py_test(
87    name = "linalg_grad_test",
88    size = "medium",
89    srcs = ["linalg_grad_test.py"],
90    shard_count = 20,
91    deps = [
92        "//tensorflow/python:array_ops",
93        "//tensorflow/python:client_testlib",
94        "//tensorflow/python:framework_for_generated_wrappers",
95        "//tensorflow/python:gradients",
96        "//tensorflow/python:linalg_ops",
97        "//tensorflow/python:math_ops",
98        "//tensorflow/python/ops/linalg",
99        "//third_party/py/numpy",
100    ],
101)
102
103cuda_py_test(
104    name = "linalg_ops_test",
105    size = "medium",
106    srcs = ["linalg_ops_test.py"],
107    shard_count = 8,
108    tags = ["no_windows_gpu"],
109    deps = [
110        "//tensorflow/python:array_ops",
111        "//tensorflow/python:client_testlib",
112        "//tensorflow/python:framework_for_generated_wrappers",
113        "//tensorflow/python:linalg_ops",
114        "//tensorflow/python:math_ops",
115        "//tensorflow/python/ops/linalg",
116        "//third_party/py/numpy",
117        "@absl_py//absl/testing:parameterized",
118    ],
119)
120
121cuda_py_test(
122    name = "linear_operator_addition_test",
123    size = "small",
124    srcs = ["linear_operator_addition_test.py"],
125    deps = [
126        "//tensorflow/python:array_ops",
127        "//tensorflow/python:client_testlib",
128        "//tensorflow/python:framework_for_generated_wrappers",
129        "//tensorflow/python:framework_test_lib",
130        "//tensorflow/python:math_ops",
131        "//tensorflow/python:platform_test",
132        "//tensorflow/python/ops/linalg",
133        "//third_party/py/numpy",
134    ],
135)
136
137cuda_py_test(
138    name = "linear_operator_adjoint_test",
139    size = "medium",
140    srcs = ["linear_operator_adjoint_test.py"],
141    shard_count = 5,
142    tags = [
143        "noasan",  # times out, b/63678675
144        "optonly",  # times out
145    ],
146    deps = [
147        "//tensorflow/python:array_ops",
148        "//tensorflow/python:client_testlib",
149        "//tensorflow/python:framework",
150        "//tensorflow/python:framework_for_generated_wrappers",
151        "//tensorflow/python:framework_test_lib",
152        "//tensorflow/python:math_ops",
153        "//tensorflow/python:platform_test",
154        "//tensorflow/python/ops/linalg",
155        "//third_party/py/numpy",
156    ],
157)
158
159cuda_py_test(
160    name = "linear_operator_algebra_test",
161    size = "small",
162    srcs = ["linear_operator_algebra_test.py"],
163    deps = [
164        "//tensorflow/python:array_ops",
165        "//tensorflow/python:client_testlib",
166        "//tensorflow/python:framework_for_generated_wrappers",
167        "//tensorflow/python:framework_test_lib",
168        "//tensorflow/python:math_ops",
169        "//tensorflow/python:platform_test",
170        "//tensorflow/python/ops/linalg",
171        "//third_party/py/numpy",
172    ],
173)
174
175cuda_py_test(
176    name = "linear_operator_block_diag_test",
177    size = "medium",
178    srcs = ["linear_operator_block_diag_test.py"],
179    shard_count = 8,
180    tags = [
181        "noasan",
182        "optonly",
183    ],
184    deps = [
185        "//tensorflow/python:array_ops",
186        "//tensorflow/python:client_testlib",
187        "//tensorflow/python:framework",
188        "//tensorflow/python:framework_for_generated_wrappers",
189        "//tensorflow/python:framework_test_lib",
190        "//tensorflow/python:math_ops",
191        "//tensorflow/python:platform_test",
192        "//tensorflow/python/ops/linalg",
193        "//third_party/py/numpy",
194    ],
195)
196
197cuda_py_test(
198    name = "linear_operator_block_lower_triangular_test",
199    size = "medium",
200    srcs = ["linear_operator_block_lower_triangular_test.py"],
201    shard_count = 8,
202    tags = [
203        "noasan",
204        "optonly",
205    ],
206    deps = [
207        "//tensorflow/python:array_ops",
208        "//tensorflow/python:client_testlib",
209        "//tensorflow/python:framework",
210        "//tensorflow/python:framework_for_generated_wrappers",
211        "//tensorflow/python:framework_test_lib",
212        "//tensorflow/python:math_ops",
213        "//tensorflow/python:platform_test",
214        "//tensorflow/python/ops/linalg",
215        "//third_party/py/numpy",
216    ],
217)
218
219cuda_py_test(
220    name = "linear_operator_composition_test",
221    size = "medium",
222    srcs = ["linear_operator_composition_test.py"],
223    shard_count = 5,
224    tags = [
225        "noasan",  # times out, b/63678675
226        "optonly",  # times out
227    ],
228    deps = [
229        "//tensorflow/python:array_ops",
230        "//tensorflow/python:client_testlib",
231        "//tensorflow/python:framework",
232        "//tensorflow/python:framework_for_generated_wrappers",
233        "//tensorflow/python:framework_test_lib",
234        "//tensorflow/python:math_ops",
235        "//tensorflow/python:platform_test",
236        "//tensorflow/python/ops/linalg",
237        "//third_party/py/numpy",
238    ],
239)
240
241cuda_py_test(
242    name = "linear_operator_circulant_test",
243    size = "medium",
244    srcs = ["linear_operator_circulant_test.py"],
245    shard_count = 15,
246    tags = [
247        "no_cuda11",  # TODO(b/197522782): reenable test after fixing.
248        "noasan",  # times out, b/63678675
249        "optonly",  # times out, b/79171797
250    ],
251    deps = [
252        "//tensorflow/python:array_ops",
253        "//tensorflow/python:client_testlib",
254        "//tensorflow/python:framework",
255        "//tensorflow/python:framework_for_generated_wrappers",
256        "//tensorflow/python:framework_test_lib",
257        "//tensorflow/python:math_ops",
258        "//tensorflow/python:platform_test",
259        "//tensorflow/python/ops/linalg",
260        "//tensorflow/python/ops/signal",
261        "//third_party/py/numpy",
262    ],
263)
264
265cuda_py_test(
266    name = "linear_operator_diag_test",
267    size = "medium",
268    srcs = ["linear_operator_diag_test.py"],
269    shard_count = 5,
270    tags = [
271        "noasan",
272        "optonly",
273    ],
274    deps = [
275        "//tensorflow/python:array_ops",
276        "//tensorflow/python:client_testlib",
277        "//tensorflow/python:framework",
278        "//tensorflow/python:framework_test_lib",
279        "//tensorflow/python:linalg_ops",
280        "//tensorflow/python:math_ops",
281        "//tensorflow/python:platform_test",
282        "//tensorflow/python:random_ops",
283        "//tensorflow/python/ops/linalg",
284    ],
285)
286
287cuda_py_test(
288    name = "linear_operator_full_matrix_test",
289    size = "medium",
290    srcs = ["linear_operator_full_matrix_test.py"],
291    shard_count = 5,
292    tags = [
293        "noasan",
294        "optonly",
295    ],
296    deps = [
297        "//tensorflow/python:array_ops",
298        "//tensorflow/python:client_testlib",
299        "//tensorflow/python:framework",
300        "//tensorflow/python:framework_for_generated_wrappers",
301        "//tensorflow/python:framework_test_lib",
302        "//tensorflow/python:platform_test",
303        "//tensorflow/python/ops/linalg",
304    ],
305)
306
307cuda_py_test(
308    name = "linear_operator_householder_test",
309    size = "medium",
310    srcs = ["linear_operator_householder_test.py"],
311    shard_count = 5,
312    tags = [
313        "noasan",
314        "optonly",
315    ],
316    deps = [
317        "//tensorflow/python:array_ops",
318        "//tensorflow/python:client_testlib",
319        "//tensorflow/python:framework",
320        "//tensorflow/python:framework_test_lib",
321        "//tensorflow/python:linalg_ops",
322        "//tensorflow/python:math_ops",
323        "//tensorflow/python:platform_test",
324        "//tensorflow/python:random_ops",
325        "//tensorflow/python/ops/linalg",
326    ],
327)
328
329cuda_py_test(
330    name = "linear_operator_identity_test",
331    size = "medium",
332    srcs = ["linear_operator_identity_test.py"],
333    shard_count = 5,
334    tags = [
335        "noasan",
336        "optonly",
337    ],
338    deps = [
339        "//tensorflow/python:array_ops",
340        "//tensorflow/python:client_testlib",
341        "//tensorflow/python:framework",
342        "//tensorflow/python:framework_test_lib",
343        "//tensorflow/python:linalg_ops",
344        "//tensorflow/python:platform_test",
345        "//tensorflow/python:random_ops",
346        "//tensorflow/python/ops/linalg",
347    ],
348)
349
350cuda_py_test(
351    name = "linear_operator_inversion_test",
352    size = "medium",
353    srcs = ["linear_operator_inversion_test.py"],
354    shard_count = 5,
355    tags = [
356        "noasan",  # times out, b/63678675
357        "optonly",  # times out
358    ],
359    deps = [
360        "//tensorflow/python:array_ops",
361        "//tensorflow/python:client_testlib",
362        "//tensorflow/python:framework",
363        "//tensorflow/python:framework_for_generated_wrappers",
364        "//tensorflow/python:framework_test_lib",
365        "//tensorflow/python:math_ops",
366        "//tensorflow/python:platform_test",
367        "//tensorflow/python/ops/linalg",
368        "//third_party/py/numpy",
369    ],
370)
371
372cuda_py_test(
373    name = "linear_operator_kronecker_test",
374    size = "medium",
375    srcs = ["linear_operator_kronecker_test.py"],
376    shard_count = 10,
377    tags = [
378        "noasan",
379        "optonly",
380    ],
381    xla_enable_strict_auto_jit = True,
382    deps = [
383        "//tensorflow/python:array_ops",
384        "//tensorflow/python:client_testlib",
385        "//tensorflow/python:framework",
386        "//tensorflow/python:framework_for_generated_wrappers",
387        "//tensorflow/python:framework_test_lib",
388        "//tensorflow/python:math_ops",
389        "//tensorflow/python:platform_test",
390        "//tensorflow/python/ops/linalg",
391        "//third_party/py/numpy",
392    ],
393)
394
395cuda_py_test(
396    name = "linear_operator_low_rank_update_test",
397    size = "medium",
398    srcs = ["linear_operator_low_rank_update_test.py"],
399    shard_count = 10,
400    tags = [
401        "noasan",  # times out
402        "optonly",
403    ],
404    deps = [
405        "//tensorflow/python:array_ops",
406        "//tensorflow/python:client_testlib",
407        "//tensorflow/python:framework",
408        "//tensorflow/python:framework_for_generated_wrappers",
409        "//tensorflow/python:framework_test_lib",
410        "//tensorflow/python:platform_test",
411        "//tensorflow/python/ops/linalg",
412    ],
413)
414
415cuda_py_test(
416    name = "linear_operator_lower_triangular_test",
417    size = "medium",
418    srcs = ["linear_operator_lower_triangular_test.py"],
419    shard_count = 4,
420    tags = [
421        "noasan",
422        "optonly",
423    ],
424    deps = [
425        "//tensorflow/python:array_ops",
426        "//tensorflow/python:client_testlib",
427        "//tensorflow/python:framework",
428        "//tensorflow/python:framework_for_generated_wrappers",
429        "//tensorflow/python:framework_test_lib",
430        "//tensorflow/python:platform_test",
431        "//tensorflow/python/ops/linalg",
432    ],
433)
434
435cuda_py_test(
436    name = "linear_operator_permutation_test",
437    size = "medium",
438    srcs = ["linear_operator_permutation_test.py"],
439    shard_count = 5,
440    tags = [
441        "noasan",
442        "optonly",
443    ],
444    xla_enable_strict_auto_jit = True,
445    deps = [
446        "//tensorflow/python:array_ops",
447        "//tensorflow/python:client_testlib",
448        "//tensorflow/python:framework",
449        "//tensorflow/python:framework_test_lib",
450        "//tensorflow/python:linalg_ops",
451        "//tensorflow/python:math_ops",
452        "//tensorflow/python:platform_test",
453        "//tensorflow/python:random_ops",
454        "//tensorflow/python/ops/linalg",
455    ],
456)
457
458cuda_py_test(
459    name = "linear_operator_test",
460    size = "small",
461    srcs = ["linear_operator_test.py"],
462    deps = [
463        "//tensorflow/python:array_ops",
464        "//tensorflow/python:client_testlib",
465        "//tensorflow/python:framework_for_generated_wrappers",
466        "//tensorflow/python:framework_test_lib",
467        "//tensorflow/python:math_ops",
468        "//tensorflow/python:platform_test",
469        "//tensorflow/python/ops/linalg",
470        "//third_party/py/numpy",
471    ],
472)
473
474cuda_py_test(
475    name = "linear_operator_toeplitz_test",
476    size = "medium",
477    srcs = ["linear_operator_toeplitz_test.py"],
478    shard_count = 5,
479    tags = [
480        "no_cuda_on_cpu_tap",  # flaky, b/135701551
481        "no_gpu",  # flaky, b/135701551
482        "noasan",  # times out, b/63678675
483        "optonly",  # times out, b/79171797
484    ],
485    deps = [
486        "//tensorflow/python:array_ops",
487        "//tensorflow/python:client_testlib",
488        "//tensorflow/python:framework",
489        "//tensorflow/python:framework_for_generated_wrappers",
490        "//tensorflow/python:framework_test_lib",
491        "//tensorflow/python:math_ops",
492        "//tensorflow/python:platform_test",
493        "//tensorflow/python/ops/linalg",
494        "//tensorflow/python/ops/signal",
495        "//third_party/py/numpy",
496    ],
497)
498
499cuda_py_test(
500    name = "linear_operator_tridiag_test",
501    size = "medium",
502    srcs = ["linear_operator_tridiag_test.py"],
503    shard_count = 5,
504    tags = [
505        "no_windows_gpu",
506        "noasan",
507        "optonly",
508    ],
509    xla_enable_strict_auto_jit = True,
510    deps = [
511        "//tensorflow/python:array_ops",
512        "//tensorflow/python:client_testlib",
513        "//tensorflow/python:framework",
514        "//tensorflow/python:framework_test_lib",
515        "//tensorflow/python:linalg_ops",
516        "//tensorflow/python:math_ops",
517        "//tensorflow/python:platform_test",
518        "//tensorflow/python:random_ops",
519        "//tensorflow/python/ops/linalg",
520    ],
521)
522
523cuda_py_test(
524    name = "linear_operator_util_test",
525    size = "medium",
526    srcs = ["linear_operator_util_test.py"],
527    shard_count = 5,
528    tags = [
529        "noasan",
530        "optonly",
531    ],
532    deps = [
533        "//tensorflow/python:client_testlib",
534        "//tensorflow/python:framework",
535        "//tensorflow/python:framework_for_generated_wrappers",
536        "//tensorflow/python:framework_test_lib",
537        "//tensorflow/python:math_ops",
538        "//tensorflow/python:platform_test",
539        "//tensorflow/python/ops/linalg",
540    ],
541)
542
543cuda_py_test(
544    name = "linear_operator_zeros_test",
545    size = "medium",
546    srcs = ["linear_operator_zeros_test.py"],
547    shard_count = 5,
548    tags = ["optonly"],  # Test is flaky without optimization.
549    deps = [
550        "//tensorflow/python:array_ops",
551        "//tensorflow/python:client_testlib",
552        "//tensorflow/python:framework",
553        "//tensorflow/python:framework_test_lib",
554        "//tensorflow/python:linalg_ops",
555        "//tensorflow/python:platform_test",
556        "//tensorflow/python:random_ops",
557        "//tensorflow/python/ops/linalg",
558    ],
559)
560
561cuda_py_test(
562    name = "lu_op_test",
563    size = "small",
564    srcs = ["lu_op_test.py"],
565    xla_tags = [
566        "no_cuda_asan",  # times out
567    ],
568    deps = [
569        "//tensorflow/python:array_ops",
570        "//tensorflow/python:client_testlib",
571        "//tensorflow/python:framework_for_generated_wrappers",
572        "//tensorflow/python:linalg_ops",
573        "//tensorflow/python:math_ops",
574        "//tensorflow/python:platform",
575        "//tensorflow/python/ops/linalg",
576        "//third_party/py/numpy",
577    ],
578)
579
580tf_py_test(
581    name = "matrix_exponential_op_test",
582    size = "medium",
583    srcs = ["matrix_exponential_op_test.py"],
584    shard_count = 16,
585    tags = ["no_windows_gpu"],
586    deps = [
587        "//tensorflow/python:client_testlib",
588        "//tensorflow/python:framework_for_generated_wrappers",
589        "//tensorflow/python:linalg_ops",
590        "//third_party/py/numpy",
591    ],
592)
593
594cuda_py_test(
595    name = "matrix_inverse_op_test",
596    size = "small",
597    srcs = ["matrix_inverse_op_test.py"],
598    tags = ["optonly"],
599    deps = [
600        "//tensorflow/python:client_testlib",
601        "//tensorflow/python:framework_for_generated_wrappers",
602        "//tensorflow/python:linalg_ops",
603        "//tensorflow/python:math_ops",
604        "//third_party/py/numpy",
605    ],
606)
607
608tf_py_test(
609    name = "matrix_logarithm_op_test",
610    size = "medium",
611    srcs = ["matrix_logarithm_op_test.py"],
612    deps = [
613        "//tensorflow/python:client_testlib",
614        "//tensorflow/python:framework_for_generated_wrappers",
615        "//tensorflow/python:linalg_ops",
616        "//third_party/py/numpy",
617    ],
618)
619
620cuda_py_test(
621    name = "matrix_solve_ls_op_test",
622    size = "medium",
623    srcs = ["matrix_solve_ls_op_test.py"],
624    tags = [
625        "noasan",  # TODO(b/337374867) fails with -fsanitize=null
626        "nomac",  # TODO(b/233829166): Breaks nightly build
627    ],
628    deps = [
629        "//tensorflow/python:array_ops",
630        "//tensorflow/python:client_testlib",
631        "//tensorflow/python:framework_for_generated_wrappers",
632        "//tensorflow/python:linalg_ops",
633        "//tensorflow/python:math_ops",
634        "//third_party/py/numpy",
635    ],
636)
637
638cuda_py_test(
639    name = "matrix_solve_op_test",
640    size = "medium",
641    srcs = ["matrix_solve_op_test.py"],
642    deps = [
643        "//tensorflow/python:array_ops",
644        "//tensorflow/python:client_testlib",
645        "//tensorflow/python:framework_for_generated_wrappers",
646        "//tensorflow/python:linalg_ops",
647        "//third_party/py/numpy",
648    ],
649)
650
651cuda_py_test(
652    name = "matrix_square_root_op_test",
653    size = "medium",
654    srcs = ["matrix_square_root_op_test.py"],
655    deps = [
656        "//tensorflow/python:client_testlib",
657        "//tensorflow/python:framework_for_generated_wrappers",
658        "//tensorflow/python:linalg_ops",
659        "//third_party/py/numpy",
660    ],
661)
662
663cuda_py_test(
664    name = "matrix_triangular_solve_op_test",
665    size = "medium",
666    srcs = ["matrix_triangular_solve_op_test.py"],
667    shard_count = 3,
668    deps = [
669        "//tensorflow/python:client_testlib",
670        "//tensorflow/python:linalg_ops",
671        "//third_party/py/numpy",
672    ],
673)
674
675cuda_py_test(
676    name = "normalize_op_test",
677    size = "medium",
678    srcs = ["normalize_op_test.py"],
679    shard_count = 20,
680    # TODO(b/117236102): Re-enable in msan build.
681    tags = [
682        "no_windows_gpu",
683        "nomsan",
684    ],
685    # TODO(b/208263392): Re-enable. tf.Squeeze op after tf.Where op doesn't reshape.
686    xla_enable_strict_auto_jit = False,
687    deps = [
688        "//tensorflow/python:client_testlib",
689        "//tensorflow/python:framework_for_generated_wrappers",
690        "//tensorflow/python:nn",
691        "//third_party/py/numpy",
692    ],
693)
694
695cuda_py_test(
696    name = "norm_op_test",
697    size = "medium",
698    srcs = ["norm_op_test.py"],
699    shard_count = 20,
700    # TODO(b/117236102): Re-enable in msan build.
701    tags = [
702        "no_windows_gpu",
703        "nomsan",
704    ],
705    # TODO(b/208263392): Re-enable. tf.Squeeze op after tf.Where op doesn't reshape.
706    xla_enable_strict_auto_jit = False,
707    deps = [
708        "//tensorflow/python:array_ops",
709        "//tensorflow/python:client_testlib",
710        "//tensorflow/python:framework_for_generated_wrappers",
711        "//tensorflow/python:linalg_ops",
712        "//third_party/py/numpy",
713    ],
714)
715
716cuda_py_test(
717    name = "qr_op_test",
718    size = "medium",
719    srcs = ["qr_op_test.py"],
720    shard_count = 20,
721    deps = [
722        "//tensorflow/python:array_ops",
723        "//tensorflow/python:client_testlib",
724        "//tensorflow/python:framework_for_generated_wrappers",
725        "//tensorflow/python:linalg_ops",
726        "//tensorflow/python:math_ops",
727        "//third_party/py/numpy",
728    ],
729)
730
731cuda_py_test(
732    name = "self_adjoint_eig_op_test",
733    size = "medium",
734    srcs = ["self_adjoint_eig_op_test.py"],
735    data = ["//tensorflow/python/kernel_tests/linalg/testdata:self_adjoint_eig_op_test_files"],
736    shard_count = 20,
737    tags = [
738        "no_windows",
739    ],
740    deps = [
741        "//tensorflow/python:array_ops",
742        "//tensorflow/python:client_testlib",
743        "//tensorflow/python:framework_for_generated_wrappers",
744        "//tensorflow/python:linalg_ops",
745        "//tensorflow/python:math_ops",
746        "//third_party/py/numpy",
747    ],
748    # TODO(b/127344411): This test passes because XLA does not actually cluster
749    # the self_adjoint_eig op.
750)
751
752cuda_py_test(
753    name = "slicing_test",
754    size = "small",
755    srcs = ["slicing_test.py"],
756    deps = [
757        "//tensorflow/python:array_ops",
758        "//tensorflow/python:client_testlib",
759        "//tensorflow/python:framework_for_generated_wrappers",
760        "//tensorflow/python:framework_test_lib",
761        "//tensorflow/python:math_ops",
762        "//tensorflow/python:platform_test",
763        "//tensorflow/python/ops/linalg",
764        "//third_party/py/numpy",
765    ],
766)
767
768cuda_py_test(
769    name = "svd_op_test",
770    size = "medium",
771    srcs = ["svd_op_test.py"],
772    shard_count = 30,
773    tags = [
774        "no_oss",  # b/117185141.
775        "nomsan",  # TODO(b/117236102): Re-enable in msan build.
776    ],
777    xla_tags = [
778        "no_cuda_asan",  # times out
779    ],
780    deps = [
781        "//tensorflow/python:array_ops",
782        "//tensorflow/python:client_testlib",
783        "//tensorflow/python:framework_for_generated_wrappers",
784        "//tensorflow/python:gradients_impl",
785        "//tensorflow/python:linalg_ops",
786        "//tensorflow/python:math_ops",
787        "//third_party/py/numpy",
788    ],
789    # TODO(b/127344411): This test passes because XLA does not actually cluster
790    # the svd op.
791)
792
793cuda_py_test(
794    name = "tridiagonal_matmul_op_test",
795    size = "medium",
796    srcs = ["tridiagonal_matmul_op_test.py"],
797    shard_count = 10,
798    deps = [
799        "//tensorflow/python:client_testlib",
800        "//tensorflow/python:framework_for_generated_wrappers",
801        "//tensorflow/python:linalg_ops",
802        "//third_party/py/numpy",
803    ],
804)
805
806cuda_py_test(
807    name = "tridiagonal_solve_op_test",
808    size = "medium",
809    srcs = ["tridiagonal_solve_op_test.py"],
810    shard_count = 12,
811    tags = [
812        "no_oss",  # TODO(b/142818120): Re-enable.
813    ],
814    deps = [
815        "//tensorflow/python:client_testlib",
816        "//tensorflow/python:framework_for_generated_wrappers",
817        "//tensorflow/python:linalg_ops",
818        "//third_party/py/numpy",
819    ],
820)
821