• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Common computation builders for XLA.
2
3load("//tensorflow:tensorflow.bzl", "filegroup")
4load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
5load("//tensorflow/compiler/xla/tests:build_defs.bzl", "generate_backend_suites", "xla_test")
6
7package(
8    default_visibility = ["//tensorflow/compiler/xla/client:friends"],
9    licenses = ["notice"],
10)
11
12# Filegroup used to collect source files for dependency checking.
13filegroup(
14    name = "c_srcs",
15    data = glob([
16        "**/*.cc",
17        "**/*.h",
18    ]),
19)
20
21# Generate test_suites for all backends, named "${backend}_tests".
22generate_backend_suites()
23
24cc_library(
25    name = "arithmetic",
26    srcs = ["arithmetic.cc"],
27    hdrs = ["arithmetic.h"],
28    deps = [
29        ":constants",
30        "//tensorflow/compiler/xla:shape_util",
31        "//tensorflow/compiler/xla:status_macros",
32        "//tensorflow/compiler/xla:types",
33        "//tensorflow/compiler/xla:xla_data_proto_cc",
34        "//tensorflow/compiler/xla/client:xla_builder",
35        "//tensorflow/compiler/xla/client:xla_computation",
36        "@com_google_absl//absl/strings",
37    ],
38)
39
40xla_test(
41    name = "arithmetic_test",
42    srcs = ["arithmetic_test.cc"],
43    deps = [
44        ":arithmetic",
45        "//tensorflow/compiler/xla:literal_util",
46        "//tensorflow/compiler/xla:shape_util",
47        "//tensorflow/compiler/xla:test",
48        "//tensorflow/compiler/xla:types",
49        "//tensorflow/compiler/xla:xla_data_proto_cc",
50        "//tensorflow/compiler/xla/client:xla_builder",
51        "//tensorflow/compiler/xla/tests:client_library_test_base",
52        "//tensorflow/compiler/xla/tests:test_macros_header",
53        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
54    ],
55)
56
57cc_library(
58    name = "comparators",
59    srcs = ["comparators.cc"],
60    hdrs = [
61        "comparators.h",
62    ],
63    deps = [
64        ":constants",
65        "//tensorflow/compiler/xla:literal_util",
66        "//tensorflow/compiler/xla:shape_util",
67        "//tensorflow/compiler/xla:types",
68        "//tensorflow/compiler/xla:xla_data_proto_cc",
69        "//tensorflow/compiler/xla/client:xla_builder",
70        "//tensorflow/compiler/xla/client:xla_computation",
71        "@com_google_absl//absl/strings",
72        "@com_google_absl//absl/types:span",
73    ],
74)
75
76xla_test(
77    name = "comparators_test",
78    srcs = ["comparators_test.cc"],
79    deps = [
80        ":comparators",
81        ":constants",
82        "//tensorflow/compiler/xla:shape_util",
83        "//tensorflow/compiler/xla:test",
84        "//tensorflow/compiler/xla:types",
85        "//tensorflow/compiler/xla:xla_data_proto_cc",
86        "//tensorflow/compiler/xla/client:xla_builder",
87        "//tensorflow/compiler/xla/tests:client_library_test_base",
88        "//tensorflow/compiler/xla/tests:test_macros_header",
89        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
90        "@com_google_absl//absl/container:inlined_vector",
91    ],
92)
93
94cc_library(
95    name = "constants",
96    srcs = ["constants.cc"],
97    hdrs = ["constants.h"],
98    deps = [
99        "//tensorflow/compiler/xla:literal_util",
100        "//tensorflow/compiler/xla:shape_util",
101        "//tensorflow/compiler/xla:types",
102        "//tensorflow/compiler/xla:util",
103        "//tensorflow/compiler/xla:xla_data_proto_cc",
104        "//tensorflow/compiler/xla/client:xla_builder",
105    ],
106)
107
108cc_library(
109    name = "broadcast",
110    srcs = ["broadcast.cc"],
111    hdrs = ["broadcast.h"],
112    deps = [
113        "//tensorflow/compiler/xla:shape_util",
114        "//tensorflow/compiler/xla:status_macros",
115        "//tensorflow/compiler/xla:types",
116        "//tensorflow/compiler/xla:util",
117        "//tensorflow/compiler/xla:xla_data_proto_cc",
118        "//tensorflow/compiler/xla/client:xla_builder",
119        "@com_google_absl//absl/algorithm:container",
120        "@com_google_absl//absl/strings",
121    ],
122)
123
124xla_test(
125    name = "constants_test",
126    srcs = ["constants_test.cc"],
127    deps = [
128        ":constants",
129        "//tensorflow/compiler/xla:test",
130        "//tensorflow/compiler/xla:types",
131        "//tensorflow/compiler/xla:xla_data_proto_cc",
132        "//tensorflow/compiler/xla/client:xla_builder",
133        "//tensorflow/compiler/xla/tests:client_library_test_base",
134        "//tensorflow/compiler/xla/tests:test_macros_header",
135        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
136    ],
137)
138
139cc_library(
140    name = "conv_grad_size_util",
141    srcs = ["conv_grad_size_util.cc"],
142    hdrs = ["conv_grad_size_util.h"],
143    deps = [
144        "//tensorflow/compiler/xla:status_macros",
145        "//tensorflow/compiler/xla/client:padding",
146        "//tensorflow/core:lib",
147    ],
148)
149
150cc_library(
151    name = "dynamic_shaped_ops",
152    srcs = ["dynamic_shaped_ops.cc"],
153    hdrs = ["dynamic_shaped_ops.h"],
154    deps = [
155        ":constants",
156        "//tensorflow/compiler/xla:literal_util",
157        "//tensorflow/compiler/xla:shape_util",
158        "//tensorflow/compiler/xla:types",
159        "//tensorflow/compiler/xla:util",
160        "//tensorflow/compiler/xla:xla_data_proto_cc",
161        "//tensorflow/compiler/xla/client:value_inference",
162        "//tensorflow/compiler/xla/client:xla_builder",
163        "@com_google_absl//absl/algorithm:container",
164        "@com_google_absl//absl/types:span",
165    ],
166)
167
168cc_library(
169    name = "loops",
170    srcs = ["loops.cc"],
171    hdrs = ["loops.h"],
172    deps = [
173        ":constants",
174        "//tensorflow/compiler/xla:shape_util",
175        "//tensorflow/compiler/xla:status_macros",
176        "//tensorflow/compiler/xla:statusor",
177        "//tensorflow/compiler/xla/client:xla_builder",
178        "//tensorflow/compiler/xla/client:xla_computation",
179        "@com_google_absl//absl/strings",
180        "@com_google_absl//absl/types:span",
181    ],
182)
183
184cc_library(
185    name = "math",
186    srcs = ["math.cc"],
187    hdrs = ["math.h"],
188    deps = [
189        ":arithmetic",
190        ":constants",
191        ":loops",
192        "//tensorflow/compiler/xla:shape_util",
193        "//tensorflow/compiler/xla:status_macros",
194        "//tensorflow/compiler/xla/client:xla_builder",
195    ],
196)
197
198xla_test(
199    name = "math_test",
200    srcs = ["math_test.cc"],
201    deps = [
202        ":constants",
203        ":math",
204        "//tensorflow/compiler/xla:literal_util",
205        "//tensorflow/compiler/xla:shape_util",
206        "//tensorflow/compiler/xla:test",
207        "//tensorflow/compiler/xla:types",
208        "//tensorflow/compiler/xla:xla_data_proto_cc",
209        "//tensorflow/compiler/xla/client:xla_builder",
210        "//tensorflow/compiler/xla/tests:client_library_test_base",
211        "//tensorflow/compiler/xla/tests:test_macros_header",
212        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
213    ],
214)
215
216cc_library(
217    name = "matrix",
218    srcs = ["matrix.cc"],
219    hdrs = ["matrix.h"],
220    deps = [
221        ":arithmetic",
222        ":constants",
223        ":slicing",
224        "//tensorflow/compiler/xla:literal",
225        "//tensorflow/compiler/xla:shape_util",
226        "//tensorflow/compiler/xla:status",
227        "//tensorflow/compiler/xla:status_macros",
228        "//tensorflow/compiler/xla:statusor",
229        "//tensorflow/compiler/xla:types",
230        "//tensorflow/compiler/xla:util",
231        "//tensorflow/compiler/xla:xla_data_proto_cc",
232        "//tensorflow/compiler/xla/client:xla_builder",
233        "@com_google_absl//absl/algorithm:container",
234        "@com_google_absl//absl/container:flat_hash_set",
235        "@com_google_absl//absl/container:inlined_vector",
236        "@com_google_absl//absl/strings",
237        "@com_google_absl//absl/types:span",
238    ],
239)
240
241xla_test(
242    name = "matrix_test",
243    srcs = ["matrix_test.cc"],
244    deps = [
245        ":constants",
246        ":matrix",
247        ":slicing",
248        "//tensorflow/compiler/xla:status",
249        "//tensorflow/compiler/xla:status_macros",
250        "//tensorflow/compiler/xla:statusor",
251        "//tensorflow/compiler/xla:test",
252        "//tensorflow/compiler/xla:types",
253        "//tensorflow/compiler/xla:xla_data_proto_cc",
254        "//tensorflow/compiler/xla/client:xla_builder",
255        "//tensorflow/compiler/xla/tests:client_library_test_base",
256        "//tensorflow/compiler/xla/tests:test_macros_header",
257        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
258        "@com_google_absl//absl/strings",
259    ],
260)
261
262cc_library(
263    name = "pooling",
264    srcs = ["pooling.cc"],
265    hdrs = ["pooling.h"],
266    deps = [
267        ":arithmetic",
268        ":constants",
269        ":conv_grad_size_util",
270        "//tensorflow/compiler/xla/client:xla_builder",
271        "@com_google_absl//absl/container:inlined_vector",
272    ],
273)
274
275xla_test(
276    name = "pooling_test",
277    srcs = ["pooling_test.cc"],
278    deps = [
279        ":pooling",
280        "//tensorflow/compiler/xla:test",
281        "//tensorflow/compiler/xla/tests:client_library_test_base",
282        "//tensorflow/compiler/xla/tests:test_macros_header",
283        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
284        "@com_google_absl//absl/container:inlined_vector",
285    ],
286)
287
288cc_library(
289    name = "prng",
290    srcs = ["prng.cc"],
291    hdrs = ["prng.h"],
292    deps = [
293        ":constants",
294        "//tensorflow/compiler/xla:util",
295        "//tensorflow/compiler/xla:xla_data_proto_cc",
296        "//tensorflow/compiler/xla/client:xla_builder",
297    ],
298)
299
300cc_library(
301    name = "qr",
302    srcs = ["qr.cc"],
303    hdrs = ["qr.h"],
304    deps = [
305        ":arithmetic",
306        ":constants",
307        ":loops",
308        ":math",
309        ":matrix",
310        ":slicing",
311        "//tensorflow/compiler/xla:literal_util",
312        "//tensorflow/compiler/xla:shape_util",
313        "//tensorflow/compiler/xla:status_macros",
314        "//tensorflow/compiler/xla:statusor",
315        "//tensorflow/compiler/xla:xla_data_proto_cc",
316        "//tensorflow/compiler/xla/client:xla_builder",
317        "//tensorflow/core:lib",
318    ],
319)
320
321xla_test(
322    name = "qr_test",
323    srcs = ["qr_test.cc"],
324    tags = ["optonly"],
325    deps = [
326        ":constants",
327        ":matrix",
328        ":qr",
329        "//tensorflow/compiler/xla:array2d",
330        "//tensorflow/compiler/xla:array3d",
331        "//tensorflow/compiler/xla:literal",
332        "//tensorflow/compiler/xla:statusor",
333        "//tensorflow/compiler/xla:test",
334        "//tensorflow/compiler/xla:types",
335        "//tensorflow/compiler/xla:xla_data_proto_cc",
336        "//tensorflow/compiler/xla/client:xla_builder",
337        "//tensorflow/compiler/xla/tests:client_library_test_base",
338        "//tensorflow/compiler/xla/tests:literal_test_util",
339        "//tensorflow/compiler/xla/tests:test_macros_header",
340        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
341        "//tensorflow/core:test",
342        "//tensorflow/core/platform:tensor_float_32_utils",
343    ],
344)
345
346cc_library(
347    name = "lu_decomposition",
348    srcs = ["lu_decomposition.cc"],
349    hdrs = ["lu_decomposition.h"],
350    deps = [
351        "//tensorflow/compiler/xla:shape_util",
352        "//tensorflow/compiler/xla:statusor",
353        "//tensorflow/compiler/xla:types",
354        "//tensorflow/compiler/xla:xla_data_proto_cc",
355        "//tensorflow/compiler/xla/client:xla_builder",
356    ],
357)
358
359cc_library(
360    name = "approx_topk",
361    srcs = ["approx_topk.cc"],
362    hdrs = ["approx_topk.h"],
363    deps = [
364        ":approx_topk_shape",
365        "//tensorflow/compiler/xla:shape_util",
366        "//tensorflow/compiler/xla:statusor",
367        "//tensorflow/compiler/xla:util",
368        "//tensorflow/compiler/xla:xla_data_proto_cc",
369        "//tensorflow/compiler/xla/client:xla_builder",
370        "//tensorflow/compiler/xla/client:xla_computation",
371        "@com_google_absl//absl/strings:str_format",
372    ],
373)
374
375cc_library(
376    name = "approx_topk_shape",
377    srcs = ["approx_topk_shape.cc"],
378    hdrs = ["approx_topk_shape.h"],
379    deps = [
380        "//tensorflow/compiler/xla:statusor",
381        "//tensorflow/compiler/xla:util",
382    ],
383)
384
385cc_library(
386    name = "slicing",
387    srcs = ["slicing.cc"],
388    hdrs = ["slicing.h"],
389    deps = [
390        ":arithmetic",
391        ":constants",
392        "//tensorflow/compiler/xla:types",
393        "//tensorflow/compiler/xla:util",
394        "//tensorflow/compiler/xla/client:xla_builder",
395        "@com_google_absl//absl/types:span",
396    ],
397)
398
399xla_test(
400    name = "slicing_test",
401    srcs = ["slicing_test.cc"],
402    deps = [
403        ":slicing",
404        "//tensorflow/compiler/xla:literal",
405        "//tensorflow/compiler/xla:literal_util",
406        "//tensorflow/compiler/xla:shape_util",
407        "//tensorflow/compiler/xla:test",
408        "//tensorflow/compiler/xla:types",
409        "//tensorflow/compiler/xla/client:xla_builder",
410        "//tensorflow/compiler/xla/tests:client_library_test_base",
411        "//tensorflow/compiler/xla/tests:test_macros_header",
412        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
413    ],
414)
415
416cc_library(
417    name = "sorting",
418    srcs = ["sorting.cc"],
419    hdrs = ["sorting.h"],
420    deps = [
421        ":comparators",
422        ":constants",
423        ":loops",
424        ":slicing",
425        "//tensorflow/compiler/xla:shape_util",
426        "//tensorflow/compiler/xla:types",
427        "//tensorflow/compiler/xla:util",
428        "//tensorflow/compiler/xla:xla_data_proto_cc",
429        "//tensorflow/compiler/xla/client:xla_builder",
430    ],
431)
432
433xla_test(
434    name = "sorting_test",
435    srcs = ["sorting_test.cc"],
436    deps = [
437        ":sorting",
438        "//tensorflow/compiler/xla:test",
439        "//tensorflow/compiler/xla:types",
440        "//tensorflow/compiler/xla/client:xla_builder",
441        "//tensorflow/compiler/xla/tests:client_library_test_base",
442        "//tensorflow/compiler/xla/tests:test_macros_header",
443        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
444    ],
445)
446
447cc_library(
448    name = "quantize",
449    hdrs = ["quantize.h"],
450    deps = [
451        ":constants",
452        "//tensorflow/compiler/xla:types",
453        "//tensorflow/compiler/xla:util",
454        "//tensorflow/compiler/xla:xla_data_proto_cc",
455        "//tensorflow/compiler/xla/client:xla_builder",
456        "//tensorflow/core:lib",
457    ],
458)
459
460xla_test(
461    name = "quantize_test",
462    srcs = ["quantize_test.cc"],
463    # TODO(b/122119490): re-enable TAP after fixing.
464    tags = [
465        "manual",
466        "notap",
467    ],
468    deps = [
469        ":quantize",
470        "//tensorflow/compiler/xla:test",
471        "//tensorflow/compiler/xla:types",
472        "//tensorflow/compiler/xla:util",
473        "//tensorflow/compiler/xla/client:xla_builder",
474        "//tensorflow/compiler/xla/tests:client_library_test_base",
475        "//tensorflow/compiler/xla/tests:test_macros_header",
476        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
477    ],
478)
479
480cc_library(
481    name = "testing",
482    srcs = ["testing.cc"],
483    hdrs = ["testing.h"],
484    deps = [
485        "//tensorflow/compiler/xla:execution_options_util",
486        "//tensorflow/compiler/xla:literal",
487        "//tensorflow/compiler/xla:shape_util",
488        "//tensorflow/compiler/xla:statusor",
489        "//tensorflow/compiler/xla:types",
490        "//tensorflow/compiler/xla:util",
491        "//tensorflow/compiler/xla:xla_data_proto_cc",
492        "//tensorflow/compiler/xla/client",
493        "//tensorflow/compiler/xla/client:global_data",
494        "//tensorflow/compiler/xla/client:xla_builder",
495        "//tensorflow/compiler/xla/client:xla_computation",
496        "//tensorflow/compiler/xla/tests:test_utils",
497        "//tensorflow/core:lib",
498        "@com_google_absl//absl/strings",
499    ],
500)
501
502cc_library(
503    name = "self_adjoint_eig",
504    srcs = ["self_adjoint_eig.cc"],
505    hdrs = ["self_adjoint_eig.h"],
506    deps = [
507        ":arithmetic",
508        ":comparators",
509        ":constants",
510        ":loops",
511        ":math",
512        ":matrix",
513        ":slicing",
514        "//tensorflow/compiler/xla:literal_util",
515        "//tensorflow/compiler/xla:shape_util",
516        "//tensorflow/compiler/xla:status_macros",
517        "//tensorflow/compiler/xla:statusor",
518        "//tensorflow/compiler/xla:util",
519        "//tensorflow/compiler/xla:xla_data_proto_cc",
520        "//tensorflow/compiler/xla/client:xla_builder",
521        "//tensorflow/core:lib",
522    ],
523)
524
525xla_test(
526    name = "self_adjoint_eig_test",
527    srcs = ["self_adjoint_eig_test.cc"],
528    real_hardware_only = True,
529    shard_count = 5,
530    tags = ["optonly"],
531    deps = [
532        ":arithmetic",
533        ":constants",
534        ":math",
535        ":matrix",
536        ":self_adjoint_eig",
537        "//tensorflow/compiler/xla:array",
538        "//tensorflow/compiler/xla:array2d",
539        "//tensorflow/compiler/xla:array3d",
540        "//tensorflow/compiler/xla:literal",
541        "//tensorflow/compiler/xla:statusor",
542        "//tensorflow/compiler/xla:test",
543        "//tensorflow/compiler/xla:xla_data_proto_cc",
544        "//tensorflow/compiler/xla/client:xla_builder",
545        "//tensorflow/compiler/xla/tests:client_library_test_base",
546        "//tensorflow/compiler/xla/tests:literal_test_util",
547        "//tensorflow/compiler/xla/tests:test_macros_header",
548        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
549        "//tensorflow/core:test",
550    ],
551)
552
553cc_library(
554    name = "svd",
555    srcs = ["svd.cc"],
556    hdrs = ["svd.h"],
557    deps = [
558        ":arithmetic",
559        ":comparators",
560        ":constants",
561        ":loops",
562        ":math",
563        ":matrix",
564        ":slicing",
565        "//tensorflow/compiler/xla:literal_util",
566        "//tensorflow/compiler/xla:shape_util",
567        "//tensorflow/compiler/xla:status_macros",
568        "//tensorflow/compiler/xla:statusor",
569        "//tensorflow/compiler/xla:xla_data_proto_cc",
570        "//tensorflow/compiler/xla/client:xla_builder",
571        "//tensorflow/core:lib",
572    ],
573)
574
575xla_test(
576    name = "svd_test",
577    srcs = ["svd_test.cc"],
578    real_hardware_only = True,
579    shard_count = 10,
580    tags = ["optonly"],
581    deps = [
582        ":arithmetic",
583        ":constants",
584        ":matrix",
585        ":slicing",
586        ":svd",
587        "//tensorflow/compiler/xla:array2d",
588        "//tensorflow/compiler/xla:array3d",
589        "//tensorflow/compiler/xla:literal",
590        "//tensorflow/compiler/xla:shape_util",
591        "//tensorflow/compiler/xla:statusor",
592        "//tensorflow/compiler/xla:test",
593        "//tensorflow/compiler/xla:xla_data_proto_cc",
594        "//tensorflow/compiler/xla/client:xla_builder",
595        "//tensorflow/compiler/xla/tests:client_library_test_base",
596        "//tensorflow/compiler/xla/tests:literal_test_util",
597        "//tensorflow/compiler/xla/tests:test_macros_header",
598        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
599        "//tensorflow/core:test",
600        "//tensorflow/core/platform:tensor_float_32_utils",
601    ],
602)
603
604cc_library(
605    name = "tridiagonal",
606    srcs = ["tridiagonal.cc"],
607    hdrs = ["tridiagonal.h"],
608    deps = [
609        ":constants",
610        ":loops",
611        ":slicing",
612        "//tensorflow/compiler/xla:shape_util",
613        "//tensorflow/compiler/xla:status",
614        "//tensorflow/compiler/xla:status_macros",
615        "//tensorflow/compiler/xla:statusor",
616        "//tensorflow/compiler/xla:xla_data_proto_cc",
617        "//tensorflow/compiler/xla/client:xla_builder",
618        "@com_google_absl//absl/types:span",
619    ],
620)
621
622xla_test(
623    name = "tridiagonal_test",
624    srcs = ["tridiagonal_test.cc"],
625    real_hardware_only = True,
626    shard_count = 10,
627    tags = ["optonly"],
628    deps = [
629        ":constants",
630        ":slicing",
631        ":tridiagonal",
632        "//tensorflow/compiler/xla:array2d",
633        "//tensorflow/compiler/xla:error_spec",
634        "//tensorflow/compiler/xla:literal",
635        "//tensorflow/compiler/xla:shape_util",
636        "//tensorflow/compiler/xla:status",
637        "//tensorflow/compiler/xla:test",
638        "//tensorflow/compiler/xla/client:xla_builder",
639        "//tensorflow/compiler/xla/tests:client_library_test_base",
640        "//tensorflow/compiler/xla/tests:literal_test_util",
641        "//tensorflow/compiler/xla/tests:test_macros_header",
642        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
643    ],
644)
645
646cc_library(
647    name = "logdet",
648    srcs = ["logdet.cc"],
649    hdrs = ["logdet.h"],
650    deps = [
651        ":arithmetic",
652        ":constants",
653        ":loops",
654        ":math",
655        ":matrix",
656        ":qr",
657        ":slicing",
658        "//tensorflow/compiler/xla:literal_util",
659        "//tensorflow/compiler/xla:shape_util",
660        "//tensorflow/compiler/xla:status_macros",
661        "//tensorflow/compiler/xla:statusor",
662        "//tensorflow/compiler/xla/client:xla_builder",
663        "//tensorflow/core:lib",
664    ],
665)
666
667xla_test(
668    name = "logdet_test",
669    srcs = ["logdet_test.cc"],
670    tags = [
671        "optonly",
672    ],
673    deps = [
674        ":logdet",
675        ":matrix",
676        "//tensorflow/compiler/xla:array2d",
677        "//tensorflow/compiler/xla:array3d",
678        "//tensorflow/compiler/xla:literal",
679        "//tensorflow/compiler/xla:statusor",
680        "//tensorflow/compiler/xla:test",
681        "//tensorflow/compiler/xla/client:xla_builder",
682        "//tensorflow/compiler/xla/tests:client_library_test_base",
683        "//tensorflow/compiler/xla/tests:literal_test_util",
684        "//tensorflow/compiler/xla/tests:test_macros_header",
685        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
686        "//tensorflow/core:test",
687    ],
688)
689