• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1package(
2    default_visibility = ["//visibility:public"],
3    licenses = ["notice"],
4)
5
6cc_library(
7    name = "add",
8    srcs = ["add.cc"],
9    hdrs = ["add.h"],
10    deps = [
11        "//tensorflow/lite/delegates/gpu/common:operations",
12        "//tensorflow/lite/delegates/gpu/common:status",
13        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
14        "@com_google_absl//absl/strings",
15    ],
16)
17
18cc_library(
19    name = "add_test_util",
20    testonly = 1,
21    srcs = ["add_test_util.cc"],
22    hdrs = ["add_test_util.h"],
23    deps = [
24        ":add",
25        "//tensorflow/lite/delegates/gpu/common:operations",
26        "//tensorflow/lite/delegates/gpu/common:status",
27        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
28    ],
29)
30
31cc_library(
32    name = "cast",
33    srcs = ["cast.cc"],
34    hdrs = ["cast.h"],
35    deps = [
36        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
37        "//tensorflow/lite/delegates/gpu/common/task:util",
38        "@com_google_absl//absl/strings",
39    ],
40)
41
42cc_library(
43    name = "cast_test_util",
44    testonly = 1,
45    srcs = ["cast_test_util.cc"],
46    hdrs = ["cast_test_util.h"],
47    deps = [
48        ":cast",
49        "//tensorflow/lite/delegates/gpu/common:status",
50        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
51    ],
52)
53
54cc_library(
55    name = "concat_xy",
56    srcs = ["concat_xy.cc"],
57    hdrs = ["concat_xy.h"],
58    deps = [
59        "//tensorflow/lite/delegates/gpu/common:operations",
60        "//tensorflow/lite/delegates/gpu/common:status",
61        "//tensorflow/lite/delegates/gpu/common:types",
62        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
63        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
64    ],
65)
66
67cc_library(
68    name = "concat_z",
69    srcs = ["concat_z.cc"],
70    hdrs = ["concat_z.h"],
71    deps = [
72        "//tensorflow/lite/delegates/gpu/common:status",
73        "//tensorflow/lite/delegates/gpu/common:types",
74        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
75        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
76    ],
77)
78
79cc_library(
80    name = "concat_test_util",
81    testonly = 1,
82    srcs = ["concat_test_util.cc"],
83    hdrs = ["concat_test_util.h"],
84    deps = [
85        ":concat_xy",
86        ":concat_z",
87        "//tensorflow/lite/delegates/gpu/common:operations",
88        "//tensorflow/lite/delegates/gpu/common:status",
89        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
90    ],
91)
92
93cc_library(
94    name = "conv_constants",
95    srcs = ["conv_constants.cc"],
96    hdrs = ["conv_constants.h"],
97    deps = [
98        "//tensorflow/lite/delegates/gpu/common:data_type",
99        "//tensorflow/lite/delegates/gpu/common:operations",
100        "//tensorflow/lite/delegates/gpu/common:shape",
101        "//tensorflow/lite/delegates/gpu/common:status",
102        "//tensorflow/lite/delegates/gpu/common:tensor",
103        "//tensorflow/lite/delegates/gpu/common:types",
104        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
105        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
106        "@com_google_absl//absl/strings",
107    ],
108)
109
110cc_library(
111    name = "conv_constants_test_util",
112    testonly = 1,
113    srcs = ["conv_constants_test_util.cc"],
114    hdrs = ["conv_constants_test_util.h"],
115    deps = [
116        ":conv_constants",
117        "//tensorflow/lite/delegates/gpu/common:operations",
118        "//tensorflow/lite/delegates/gpu/common:status",
119        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
120    ],
121)
122
123cc_library(
124    name = "conv_metal_simd",
125    srcs = ["conv_metal_simd.cc"],
126    hdrs = ["conv_metal_simd.h"],
127    deps = [
128        "//tensorflow/lite/delegates/gpu/common:data_type",
129        "//tensorflow/lite/delegates/gpu/common:gpu_info",
130        "//tensorflow/lite/delegates/gpu/common:operations",
131        "//tensorflow/lite/delegates/gpu/common:shape",
132        "//tensorflow/lite/delegates/gpu/common:types",
133        "//tensorflow/lite/delegates/gpu/common:util",
134        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
135        "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
136    ],
137)
138
139cc_library(
140    name = "conv_generic",
141    srcs = ["conv_generic.cc"],
142    hdrs = ["conv_generic.h"],
143    deps = [
144        "//tensorflow/lite/delegates/gpu/common:data_type",
145        "//tensorflow/lite/delegates/gpu/common:operations",
146        "//tensorflow/lite/delegates/gpu/common:shape",
147        "//tensorflow/lite/delegates/gpu/common:status",
148        "//tensorflow/lite/delegates/gpu/common:tensor",
149        "//tensorflow/lite/delegates/gpu/common:types",
150        "//tensorflow/lite/delegates/gpu/common:winograd_util",
151        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
152        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
153        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
154        "//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
155        "//tensorflow/lite/delegates/gpu/common/task:util",
156        "//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
157        "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
158        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
159        "@com_google_absl//absl/strings",
160    ],
161)
162
163cc_library(
164    name = "conv_generic_test_util",
165    testonly = 1,
166    srcs = ["conv_generic_test_util.cc"],
167    hdrs = ["conv_generic_test_util.h"],
168    deps = [
169        ":conv_generic",
170        "//tensorflow/lite/delegates/gpu/common:operations",
171        "//tensorflow/lite/delegates/gpu/common:status",
172        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
173    ],
174)
175
176cc_library(
177    name = "conv_weights_converter",
178    srcs = ["conv_weights_converter.cc"],
179    hdrs = ["conv_weights_converter.h"],
180    deps = [
181        "//tensorflow/lite/delegates/gpu/common:status",
182        "//tensorflow/lite/delegates/gpu/common:types",
183        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
184        "//tensorflow/lite/delegates/gpu/common/task:util",
185        "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
186    ],
187)
188
189cc_library(
190    name = "conv_weights_converter_test_util",
191    testonly = 1,
192    srcs = ["conv_weights_converter_test_util.cc"],
193    hdrs = ["conv_weights_converter_test_util.h"],
194    deps = [
195        ":conv_weights_converter",
196        "//tensorflow/lite/delegates/gpu/common:status",
197        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
198        "//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
199    ],
200)
201
202cc_library(
203    name = "convolution_transposed",
204    srcs = ["convolution_transposed.cc"],
205    hdrs = ["convolution_transposed.h"],
206    deps = [
207        "//tensorflow/lite/delegates/gpu/common:data_type",
208        "//tensorflow/lite/delegates/gpu/common:operations",
209        "//tensorflow/lite/delegates/gpu/common:shape",
210        "//tensorflow/lite/delegates/gpu/common:status",
211        "//tensorflow/lite/delegates/gpu/common:tensor",
212        "//tensorflow/lite/delegates/gpu/common:types",
213        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
214        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
215        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
216        "//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
217        "//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
218        "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
219        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
220        "@com_google_absl//absl/strings",
221    ],
222)
223
224cc_library(
225    name = "convolution_transposed_test_util",
226    testonly = 1,
227    srcs = ["convolution_transposed_test_util.cc"],
228    hdrs = ["convolution_transposed_test_util.h"],
229    deps = [
230        ":convolution_transposed",
231        "//tensorflow/lite/delegates/gpu/common:operations",
232        "//tensorflow/lite/delegates/gpu/common:status",
233        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
234    ],
235)
236
237cc_library(
238    name = "convolution_transposed_3x3",
239    srcs = ["convolution_transposed_3x3.cc"],
240    hdrs = ["convolution_transposed_3x3.h"],
241    deps = [
242        "//tensorflow/lite/delegates/gpu/common:data_type",
243        "//tensorflow/lite/delegates/gpu/common:operations",
244        "//tensorflow/lite/delegates/gpu/common:shape",
245        "//tensorflow/lite/delegates/gpu/common:status",
246        "//tensorflow/lite/delegates/gpu/common:tensor",
247        "//tensorflow/lite/delegates/gpu/common:types",
248        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
249        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
250        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
251        "//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
252        "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
253        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
254    ],
255)
256
257cc_library(
258    name = "convolution_transposed_3x3_test_util",
259    testonly = 1,
260    srcs = ["convolution_transposed_3x3_test_util.cc"],
261    hdrs = ["convolution_transposed_3x3_test_util.h"],
262    deps = [
263        ":convolution_transposed_3x3",
264        "//tensorflow/lite/delegates/gpu/common:operations",
265        "//tensorflow/lite/delegates/gpu/common:status",
266        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
267    ],
268)
269
270cc_library(
271    name = "convolution_transposed_3x3_thin",
272    srcs = ["convolution_transposed_3x3_thin.cc"],
273    hdrs = ["convolution_transposed_3x3_thin.h"],
274    deps = [
275        "//tensorflow/lite/delegates/gpu/common:data_type",
276        "//tensorflow/lite/delegates/gpu/common:operations",
277        "//tensorflow/lite/delegates/gpu/common:precision",
278        "//tensorflow/lite/delegates/gpu/common:shape",
279        "//tensorflow/lite/delegates/gpu/common:status",
280        "//tensorflow/lite/delegates/gpu/common:tensor",
281        "//tensorflow/lite/delegates/gpu/common:types",
282        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
283        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
284        "//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
285        "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
286        "@com_google_absl//absl/strings",
287    ],
288)
289
290cc_library(
291    name = "convolution_transposed_3x3_thin_test_util",
292    testonly = 1,
293    srcs = ["convolution_transposed_3x3_thin_test_util.cc"],
294    hdrs = ["convolution_transposed_3x3_thin_test_util.h"],
295    deps = [
296        ":convolution_transposed_3x3_thin",
297        "//tensorflow/lite/delegates/gpu/common:operations",
298        "//tensorflow/lite/delegates/gpu/common:status",
299        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
300    ],
301)
302
303cc_library(
304    name = "convolution_transposed_4x4",
305    srcs = ["convolution_transposed_4x4.cc"],
306    hdrs = ["convolution_transposed_4x4.h"],
307    deps = [
308        "//tensorflow/lite/delegates/gpu/common:data_type",
309        "//tensorflow/lite/delegates/gpu/common:operations",
310        "//tensorflow/lite/delegates/gpu/common:shape",
311        "//tensorflow/lite/delegates/gpu/common:status",
312        "//tensorflow/lite/delegates/gpu/common:tensor",
313        "//tensorflow/lite/delegates/gpu/common:types",
314        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
315        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
316        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
317        "//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
318        "//tensorflow/lite/delegates/gpu/common/task:weights_layout",
319        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
320    ],
321)
322
323cc_library(
324    name = "convolution_transposed_4x4_test_util",
325    testonly = 1,
326    srcs = ["convolution_transposed_4x4_test_util.cc"],
327    hdrs = ["convolution_transposed_4x4_test_util.h"],
328    deps = [
329        ":convolution_transposed_4x4",
330        "//tensorflow/lite/delegates/gpu/common:operations",
331        "//tensorflow/lite/delegates/gpu/common:status",
332        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
333    ],
334)
335
336cc_library(
337    name = "convolution_transposed_thin",
338    srcs = ["convolution_transposed_thin.cc"],
339    hdrs = ["convolution_transposed_thin.h"],
340    deps = [
341        "//tensorflow/lite/delegates/gpu/common:data_type",
342        "//tensorflow/lite/delegates/gpu/common:operations",
343        "//tensorflow/lite/delegates/gpu/common:shape",
344        "//tensorflow/lite/delegates/gpu/common:status",
345        "//tensorflow/lite/delegates/gpu/common:tensor",
346        "//tensorflow/lite/delegates/gpu/common:types",
347        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
348        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
349        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
350        "//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
351        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
352    ],
353)
354
355cc_library(
356    name = "convolution_transposed_thin_test_util",
357    testonly = 1,
358    srcs = ["convolution_transposed_thin_test_util.cc"],
359    hdrs = ["convolution_transposed_thin_test_util.h"],
360    deps = [
361        ":convolution_transposed_thin",
362        "//tensorflow/lite/delegates/gpu/common:operations",
363        "//tensorflow/lite/delegates/gpu/common:status",
364        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
365    ],
366)
367
368cc_library(
369    name = "cumsum",
370    srcs = ["cumsum.cc"],
371    hdrs = ["cumsum.h"],
372    deps = [
373        "//tensorflow/lite/delegates/gpu/common:operations",
374        "//tensorflow/lite/delegates/gpu/common:shape",
375        "//tensorflow/lite/delegates/gpu/common:status",
376        "//tensorflow/lite/delegates/gpu/common:types",
377        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
378    ],
379)
380
381cc_library(
382    name = "cumsum_test_util",
383    testonly = 1,
384    srcs = ["cumsum_test_util.cc"],
385    hdrs = ["cumsum_test_util.h"],
386    deps = [
387        ":cumsum",
388        "//tensorflow/lite/delegates/gpu/common:status",
389        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
390        "@com_google_absl//absl/status",
391    ],
392)
393
394cc_library(
395    name = "depthwise_conv",
396    srcs = ["depthwise_conv.cc"],
397    hdrs = ["depthwise_conv.h"],
398    deps = [
399        "//tensorflow/lite/delegates/gpu/common:data_type",
400        "//tensorflow/lite/delegates/gpu/common:operations",
401        "//tensorflow/lite/delegates/gpu/common:shape",
402        "//tensorflow/lite/delegates/gpu/common:status",
403        "//tensorflow/lite/delegates/gpu/common:tensor",
404        "//tensorflow/lite/delegates/gpu/common:types",
405        "//tensorflow/lite/delegates/gpu/common:util",
406        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
407        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
408        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
409        "//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
410        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
411    ],
412)
413
414cc_library(
415    name = "depthwise_conv_test_util",
416    testonly = 1,
417    srcs = ["depthwise_conv_test_util.cc"],
418    hdrs = ["depthwise_conv_test_util.h"],
419    deps = [
420        ":depthwise_conv",
421        "//tensorflow/lite/delegates/gpu/common:operations",
422        "//tensorflow/lite/delegates/gpu/common:status",
423        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
424    ],
425)
426
427cc_library(
428    name = "depthwise_conv_3x3",
429    srcs = ["depthwise_conv_3x3.cc"],
430    hdrs = ["depthwise_conv_3x3.h"],
431    deps = [
432        "//tensorflow/lite/delegates/gpu/common:data_type",
433        "//tensorflow/lite/delegates/gpu/common:operations",
434        "//tensorflow/lite/delegates/gpu/common:shape",
435        "//tensorflow/lite/delegates/gpu/common:status",
436        "//tensorflow/lite/delegates/gpu/common:tensor",
437        "//tensorflow/lite/delegates/gpu/common:types",
438        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
439        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
440        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
441        "//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
442        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
443        "@com_google_absl//absl/strings",
444    ],
445)
446
447cc_library(
448    name = "depthwise_conv_3x3_test_util",
449    testonly = 1,
450    srcs = ["depthwise_conv_3x3_test_util.cc"],
451    hdrs = ["depthwise_conv_3x3_test_util.h"],
452    deps = [
453        ":depthwise_conv_3x3",
454        "//tensorflow/lite/delegates/gpu/common:operations",
455        "//tensorflow/lite/delegates/gpu/common:status",
456        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
457    ],
458)
459
460cc_library(
461    name = "depthwise_conv_3x3_stride_h2",
462    srcs = ["depthwise_conv_3x3_stride_h2.cc"],
463    hdrs = ["depthwise_conv_3x3_stride_h2.h"],
464    deps = [
465        "//tensorflow/lite/delegates/gpu/common:operations",
466        "//tensorflow/lite/delegates/gpu/common:status",
467        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
468        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
469        "//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
470        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
471    ],
472)
473
474cc_library(
475    name = "depthwise_conv_3x3_stride_h2_test_util",
476    testonly = 1,
477    srcs = ["depthwise_conv_3x3_stride_h2_test_util.cc"],
478    hdrs = ["depthwise_conv_3x3_stride_h2_test_util.h"],
479    deps = [
480        ":depthwise_conv_3x3_stride_h2",
481        "//tensorflow/lite/delegates/gpu/common:operations",
482        "//tensorflow/lite/delegates/gpu/common:status",
483        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
484    ],
485)
486
487cc_library(
488    name = "elementwise",
489    srcs = ["elementwise.cc"],
490    hdrs = ["elementwise.h"],
491    deps = [
492        "//tensorflow/lite/delegates/gpu/common:operations",
493        "//tensorflow/lite/delegates/gpu/common:status",
494        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
495        "@com_google_absl//absl/strings",
496    ],
497)
498
499cc_library(
500    name = "elementwise_test_util",
501    testonly = 1,
502    srcs = ["elementwise_test_util.cc"],
503    hdrs = ["elementwise_test_util.h"],
504    deps = [
505        ":elementwise",
506        "//tensorflow/lite/delegates/gpu/common:operations",
507        "//tensorflow/lite/delegates/gpu/common:status",
508        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
509    ],
510)
511
512cc_library(
513    name = "fully_connected",
514    srcs = ["fully_connected.cc"],
515    hdrs = ["fully_connected.h"],
516    deps = [
517        "//tensorflow/lite/delegates/gpu/common:data_type",
518        "//tensorflow/lite/delegates/gpu/common:operations",
519        "//tensorflow/lite/delegates/gpu/common:shape",
520        "//tensorflow/lite/delegates/gpu/common:tensor",
521        "//tensorflow/lite/delegates/gpu/common:types",
522        "//tensorflow/lite/delegates/gpu/common:util",
523        "//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
524        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
525        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
526        "//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
527        "@com_google_absl//absl/memory",
528    ],
529)
530
531cc_library(
532    name = "fully_connected_test_util",
533    testonly = 1,
534    srcs = ["fully_connected_test_util.cc"],
535    hdrs = ["fully_connected_test_util.h"],
536    deps = [
537        ":fully_connected",
538        "//tensorflow/lite/delegates/gpu/common:operations",
539        "//tensorflow/lite/delegates/gpu/common:status",
540        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
541    ],
542)
543
544cc_library(
545    name = "gather",
546    srcs = ["gather.cc"],
547    hdrs = ["gather.h"],
548    deps = [
549        "//tensorflow/lite/delegates/gpu/common:operations",
550        "//tensorflow/lite/delegates/gpu/common:status",
551        "//tensorflow/lite/delegates/gpu/common:types",
552        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
553    ],
554)
555
556cc_library(
557    name = "gather_test_util",
558    testonly = 1,
559    srcs = ["gather_test_util.cc"],
560    hdrs = ["gather_test_util.h"],
561    deps = [
562        ":gather",
563        "//tensorflow/lite/delegates/gpu/common:operations",
564        "//tensorflow/lite/delegates/gpu/common:status",
565        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
566        "@com_google_googletest//:gtest",
567    ],
568)
569
570cc_library(
571    name = "lstm",
572    srcs = ["lstm.cc"],
573    hdrs = ["lstm.h"],
574    deps = [
575        "//tensorflow/lite/delegates/gpu/common:operations",
576        "//tensorflow/lite/delegates/gpu/common:status",
577        "//tensorflow/lite/delegates/gpu/common:types",
578        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
579        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
580    ],
581)
582
583cc_library(
584    name = "lstm_test_util",
585    testonly = 1,
586    srcs = ["lstm_test_util.cc"],
587    hdrs = ["lstm_test_util.h"],
588    deps = [
589        ":lstm",
590        "//tensorflow/lite/delegates/gpu/common:operations",
591        "//tensorflow/lite/delegates/gpu/common:status",
592        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
593        "@com_google_googletest//:gtest",
594    ],
595)
596
597cc_library(
598    name = "max_unpooling",
599    srcs = ["max_unpooling.cc"],
600    hdrs = ["max_unpooling.h"],
601    deps = [
602        "//tensorflow/lite/delegates/gpu/common:operations",
603        "//tensorflow/lite/delegates/gpu/common:status",
604        "//tensorflow/lite/delegates/gpu/common:types",
605        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
606    ],
607)
608
609cc_library(
610    name = "max_unpooling_test_util",
611    testonly = 1,
612    srcs = ["max_unpooling_test_util.cc"],
613    hdrs = ["max_unpooling_test_util.h"],
614    deps = [
615        ":max_unpooling",
616        "//tensorflow/lite/delegates/gpu/common:operations",
617        "//tensorflow/lite/delegates/gpu/common:status",
618        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
619    ],
620)
621
622cc_library(
623    name = "mean_stddev_normalization",
624    srcs = ["mean_stddev_normalization.cc"],
625    hdrs = ["mean_stddev_normalization.h"],
626    deps = [
627        "//tensorflow/lite/delegates/gpu/common:model",
628        "//tensorflow/lite/delegates/gpu/common:operations",
629        "//tensorflow/lite/delegates/gpu/common:status",
630        "//tensorflow/lite/delegates/gpu/common:types",
631        "//tensorflow/lite/delegates/gpu/common:util",
632        "//tensorflow/lite/delegates/gpu/common/selectors:subgraph",
633        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
634        "@com_google_absl//absl/strings",
635    ],
636)
637
638cc_library(
639    name = "mean_stddev_normalization_test_util",
640    testonly = 1,
641    srcs = ["mean_stddev_normalization_test_util.cc"],
642    hdrs = ["mean_stddev_normalization_test_util.h"],
643    deps = [
644        ":mean_stddev_normalization",
645        "//tensorflow/lite/delegates/gpu/common:operations",
646        "//tensorflow/lite/delegates/gpu/common:status",
647        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
648    ],
649)
650
651cc_library(
652    name = "one_hot",
653    srcs = ["one_hot.cc"],
654    hdrs = ["one_hot.h"],
655    deps = [
656        "//tensorflow/lite/delegates/gpu/common:operations",
657        "//tensorflow/lite/delegates/gpu/common:status",
658        "//tensorflow/lite/delegates/gpu/common:types",
659        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
660        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
661    ],
662)
663
664cc_library(
665    name = "one_hot_test_util",
666    testonly = 1,
667    srcs = ["one_hot_test_util.cc"],
668    hdrs = ["one_hot_test_util.h"],
669    deps = [
670        ":one_hot",
671        "//tensorflow/lite/delegates/gpu/common:operations",
672        "//tensorflow/lite/delegates/gpu/common:status",
673        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
674    ],
675)
676
677cc_library(
678    name = "padding",
679    srcs = ["padding.cc"],
680    hdrs = ["padding.h"],
681    deps = [
682        "//tensorflow/lite/delegates/gpu/common:operations",
683        "//tensorflow/lite/delegates/gpu/common:status",
684        "//tensorflow/lite/delegates/gpu/common:types",
685        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
686        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
687    ],
688)
689
690cc_library(
691    name = "padding_test_util",
692    testonly = 1,
693    srcs = ["padding_test_util.cc"],
694    hdrs = ["padding_test_util.h"],
695    deps = [
696        ":padding",
697        "//tensorflow/lite/delegates/gpu/common:operations",
698        "//tensorflow/lite/delegates/gpu/common:status",
699        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
700    ],
701)
702
703cc_library(
704    name = "pooling",
705    srcs = ["pooling.cc"],
706    hdrs = ["pooling.h"],
707    deps = [
708        "//tensorflow/lite/delegates/gpu/common:gpu_info",
709        "//tensorflow/lite/delegates/gpu/common:operations",
710        "//tensorflow/lite/delegates/gpu/common:shape",
711        "//tensorflow/lite/delegates/gpu/common:status",
712        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
713        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
714        "//tensorflow/lite/delegates/gpu/common/task:util",
715    ],
716)
717
718cc_library(
719    name = "pooling_test_util",
720    testonly = 1,
721    srcs = ["pooling_test_util.cc"],
722    hdrs = ["pooling_test_util.h"],
723    deps = [
724        ":pooling",
725        "//tensorflow/lite/delegates/gpu/common:operations",
726        "//tensorflow/lite/delegates/gpu/common:status",
727        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
728    ],
729)
730
731cc_library(
732    name = "prelu",
733    srcs = ["prelu.cc"],
734    hdrs = ["prelu.h"],
735    deps = [
736        "//tensorflow/lite/delegates/gpu/common:data_type",
737        "//tensorflow/lite/delegates/gpu/common:operations",
738        "//tensorflow/lite/delegates/gpu/common:status",
739        "//tensorflow/lite/delegates/gpu/common:tensor",
740        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
741        "@com_google_absl//absl/strings",
742        "@com_google_absl//absl/types:variant",
743    ],
744)
745
746cc_library(
747    name = "prelu_test_util",
748    testonly = 1,
749    srcs = ["prelu_test_util.cc"],
750    hdrs = ["prelu_test_util.h"],
751    deps = [
752        ":prelu",
753        "//tensorflow/lite/delegates/gpu/common:operations",
754        "//tensorflow/lite/delegates/gpu/common:status",
755        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
756    ],
757)
758
759cc_library(
760    name = "quantize_and_dequantize",
761    srcs = ["quantize_and_dequantize.cc"],
762    hdrs = ["quantize_and_dequantize.h"],
763    deps = [
764        "//tensorflow/lite/delegates/gpu/common:data_type",
765        "//tensorflow/lite/delegates/gpu/common:operations",
766        "//tensorflow/lite/delegates/gpu/common:status",
767        "//tensorflow/lite/delegates/gpu/common:tensor",
768        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
769        "@com_google_absl//absl/strings",
770        "@com_google_absl//absl/types:variant",
771    ],
772)
773
774cc_library(
775    name = "quantize_and_dequantize_test_util",
776    testonly = 1,
777    srcs = ["quantize_and_dequantize_test_util.cc"],
778    hdrs = ["quantize_and_dequantize_test_util.h"],
779    deps = [
780        ":quantize_and_dequantize",
781        "//tensorflow/lite/delegates/gpu/common:operations",
782        "//tensorflow/lite/delegates/gpu/common:status",
783        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
784        "//tensorflow/lite/kernels/internal:quantization_util",
785    ],
786)
787
788cc_library(
789    name = "reduce",
790    srcs = ["reduce.cc"],
791    hdrs = ["reduce.h"],
792    deps = [
793        "//tensorflow/lite/delegates/gpu/common:kernel_info",
794        "//tensorflow/lite/delegates/gpu/common:operations",
795        "//tensorflow/lite/delegates/gpu/common:status",
796        "//tensorflow/lite/delegates/gpu/common:types",
797        "//tensorflow/lite/delegates/gpu/common:util",
798        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
799        "//tensorflow/lite/delegates/gpu/common/task:util",
800        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
801        "@com_google_absl//absl/strings",
802    ],
803)
804
805cc_library(
806    name = "reduce_test_util",
807    testonly = 1,
808    srcs = ["reduce_test_util.cc"],
809    hdrs = ["reduce_test_util.h"],
810    deps = [
811        ":reduce",
812        "//tensorflow/lite/delegates/gpu/common:operations",
813        "//tensorflow/lite/delegates/gpu/common:status",
814        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
815    ],
816)
817
818cc_library(
819    name = "relu",
820    srcs = ["relu.cc"],
821    hdrs = ["relu.h"],
822    deps = [
823        "//tensorflow/lite/delegates/gpu/common:operations",
824        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
825        "@com_google_absl//absl/strings",
826    ],
827)
828
829cc_library(
830    name = "relu_test_util",
831    testonly = 1,
832    srcs = ["relu_test_util.cc"],
833    hdrs = ["relu_test_util.h"],
834    deps = [
835        ":relu",
836        "//tensorflow/lite/delegates/gpu/common:operations",
837        "//tensorflow/lite/delegates/gpu/common:status",
838        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
839    ],
840)
841
842cc_library(
843    name = "resampler",
844    srcs = ["resampler.cc"],
845    hdrs = ["resampler.h"],
846    deps = [
847        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
848    ],
849)
850
851cc_library(
852    name = "resampler_test_util",
853    testonly = 1,
854    srcs = ["resampler_test_util.cc"],
855    hdrs = ["resampler_test_util.h"],
856    deps = [
857        ":resampler",
858        "//tensorflow/lite/delegates/gpu/common:operations",
859        "//tensorflow/lite/delegates/gpu/common:status",
860        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
861    ],
862)
863
864cc_library(
865    name = "reshape",
866    srcs = ["reshape.cc"],
867    hdrs = ["reshape.h"],
868    deps = [
869        "//tensorflow/lite/delegates/gpu/common:status",
870        "//tensorflow/lite/delegates/gpu/common:types",
871        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
872        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
873    ],
874)
875
876cc_library(
877    name = "reshapex4",
878    srcs = ["reshapex4.cc"],
879    hdrs = ["reshapex4.h"],
880    deps = [
881        "//tensorflow/lite/delegates/gpu/common:status",
882        "//tensorflow/lite/delegates/gpu/common:types",
883        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
884        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
885    ],
886)
887
888cc_library(
889    name = "reshape_test_util",
890    testonly = 1,
891    srcs = ["reshape_test_util.cc"],
892    hdrs = ["reshape_test_util.h"],
893    deps = [
894        ":reshape",
895        ":reshapex4",
896        "//tensorflow/lite/delegates/gpu/common:operations",
897        "//tensorflow/lite/delegates/gpu/common:status",
898        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
899    ],
900)
901
902cc_library(
903    name = "resize",
904    srcs = ["resize.cc"],
905    hdrs = ["resize.h"],
906    deps = [
907        "//tensorflow/lite/delegates/gpu/common:operations",
908        "//tensorflow/lite/delegates/gpu/common:status",
909        "//tensorflow/lite/delegates/gpu/common:types",
910        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
911        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
912    ],
913)
914
915cc_library(
916    name = "resize_test_util",
917    testonly = 1,
918    srcs = ["resize_test_util.cc"],
919    hdrs = ["resize_test_util.h"],
920    deps = [
921        ":resize",
922        "//tensorflow/lite/delegates/gpu/common:operations",
923        "//tensorflow/lite/delegates/gpu/common:status",
924        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
925    ],
926)
927
928cc_library(
929    name = "select_v2",
930    srcs = ["select_v2.cc"],
931    hdrs = ["select_v2.h"],
932    deps = [
933        "//tensorflow/lite/delegates/gpu/common:operations",
934        "//tensorflow/lite/delegates/gpu/common:status",
935        "//tensorflow/lite/delegates/gpu/common:types",
936        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
937        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
938    ],
939)
940
941cc_library(
942    name = "select_v2_test_util",
943    testonly = 1,
944    srcs = ["select_v2_test_util.cc"],
945    hdrs = ["select_v2_test_util.h"],
946    deps = [
947        ":select_v2",
948        "//tensorflow/lite/delegates/gpu/common:operations",
949        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
950        "@com_google_absl//absl/status",
951    ],
952)
953
954cc_library(
955    name = "softmax",
956    srcs = ["softmax.cc"],
957    hdrs = ["softmax.h"],
958    deps = [
959        "//tensorflow/lite/delegates/gpu/common:status",
960        "//tensorflow/lite/delegates/gpu/common:types",
961        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
962        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
963        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
964    ],
965)
966
967cc_library(
968    name = "softmax1x1",
969    srcs = ["softmax1x1.cc"],
970    hdrs = ["softmax1x1.h"],
971    deps = [
972        "//tensorflow/lite/delegates/gpu/common:status",
973        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
974        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
975        "//tensorflow/lite/delegates/gpu/common/task:util",
976    ],
977)
978
979cc_library(
980    name = "softmax_test_util",
981    testonly = 1,
982    srcs = ["softmax_test_util.cc"],
983    hdrs = ["softmax_test_util.h"],
984    deps = [
985        ":softmax",
986        ":softmax1x1",
987        "//tensorflow/lite/delegates/gpu/common:operations",
988        "//tensorflow/lite/delegates/gpu/common:status",
989        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
990    ],
991)
992
993cc_library(
994    name = "space_to_depth",
995    srcs = ["space_to_depth.cc"],
996    hdrs = ["space_to_depth.h"],
997    deps = [
998        "//tensorflow/lite/delegates/gpu/common:operations",
999        "//tensorflow/lite/delegates/gpu/common:status",
1000        "//tensorflow/lite/delegates/gpu/common:types",
1001        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
1002        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
1003    ],
1004)
1005
1006cc_library(
1007    name = "space_to_depth_test_util",
1008    testonly = 1,
1009    srcs = ["space_to_depth_test_util.cc"],
1010    hdrs = ["space_to_depth_test_util.h"],
1011    deps = [
1012        ":space_to_depth",
1013        "//tensorflow/lite/delegates/gpu/common:operations",
1014        "//tensorflow/lite/delegates/gpu/common:status",
1015        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
1016    ],
1017)
1018
1019cc_library(
1020    name = "split",
1021    srcs = ["split.cc"],
1022    hdrs = ["split.h"],
1023    deps = [
1024        "//tensorflow/lite/delegates/gpu/common:operations",
1025        "//tensorflow/lite/delegates/gpu/common:types",
1026        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
1027    ],
1028)
1029
1030cc_library(
1031    name = "split_test_util",
1032    testonly = 1,
1033    srcs = ["split_test_util.cc"],
1034    hdrs = ["split_test_util.h"],
1035    deps = [
1036        ":split",
1037        "//tensorflow/lite/delegates/gpu/common:operations",
1038        "//tensorflow/lite/delegates/gpu/common:status",
1039        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
1040    ],
1041)
1042
1043cc_library(
1044    name = "strided_slice",
1045    srcs = ["strided_slice.cc"],
1046    hdrs = ["strided_slice.h"],
1047    deps = [
1048        "//tensorflow/lite/delegates/gpu/common:operations",
1049        "//tensorflow/lite/delegates/gpu/common:types",
1050        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
1051        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
1052    ],
1053)
1054
1055cc_library(
1056    name = "strided_slice_test_util",
1057    testonly = 1,
1058    srcs = ["strided_slice_test_util.cc"],
1059    hdrs = ["strided_slice_test_util.h"],
1060    deps = [
1061        ":strided_slice",
1062        "//tensorflow/lite/delegates/gpu/common:operations",
1063        "//tensorflow/lite/delegates/gpu/common:status",
1064        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
1065    ],
1066)
1067
1068cc_library(
1069    name = "tile",
1070    srcs = ["tile.cc"],
1071    hdrs = ["tile.h"],
1072    deps = [
1073        "//tensorflow/lite/delegates/gpu/common:operations",
1074        "//tensorflow/lite/delegates/gpu/common:status",
1075        "//tensorflow/lite/delegates/gpu/common:types",
1076        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
1077    ],
1078)
1079
1080cc_library(
1081    name = "tile_test_util",
1082    testonly = 1,
1083    srcs = ["tile_test_util.cc"],
1084    hdrs = ["tile_test_util.h"],
1085    deps = [
1086        ":tile",
1087        "//tensorflow/lite/delegates/gpu/common:operations",
1088        "//tensorflow/lite/delegates/gpu/common:status",
1089        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
1090    ],
1091)
1092
1093cc_library(
1094    name = "transpose",
1095    srcs = ["transpose.cc"],
1096    hdrs = ["transpose.h"],
1097    deps = [
1098        "//tensorflow/lite/delegates/gpu/common:operations",
1099        "//tensorflow/lite/delegates/gpu/common:types",
1100        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
1101        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
1102        "@com_google_absl//absl/strings",
1103    ],
1104)
1105
1106cc_library(
1107    name = "transpose_test_util",
1108    testonly = 1,
1109    srcs = ["transpose_test_util.cc"],
1110    hdrs = ["transpose_test_util.h"],
1111    deps = [
1112        ":transpose",
1113        "//tensorflow/lite/delegates/gpu/common:operations",
1114        "//tensorflow/lite/delegates/gpu/common:status",
1115        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
1116    ],
1117)
1118
1119cc_library(
1120    name = "winograd",
1121    srcs = ["winograd.cc"],
1122    hdrs = ["winograd.h"],
1123    deps = [
1124        "//tensorflow/lite/delegates/gpu/common:data_type",
1125        "//tensorflow/lite/delegates/gpu/common:operations",
1126        "//tensorflow/lite/delegates/gpu/common:shape",
1127        "//tensorflow/lite/delegates/gpu/common:status",
1128        "//tensorflow/lite/delegates/gpu/common:winograd_util",
1129        "//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
1130        "//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
1131        "//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
1132        "@com_google_absl//absl/strings:str_format",
1133    ],
1134)
1135
1136cc_library(
1137    name = "winograd_test_util",
1138    testonly = 1,
1139    srcs = ["winograd_test_util.cc"],
1140    hdrs = ["winograd_test_util.h"],
1141    deps = [
1142        ":winograd",
1143        "//tensorflow/lite/delegates/gpu/common:operations",
1144        "//tensorflow/lite/delegates/gpu/common:status",
1145        "//tensorflow/lite/delegates/gpu/common:winograd_util",
1146        "//tensorflow/lite/delegates/gpu/common/task:testing_util",
1147    ],
1148)
1149