• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite_combined")
2load(
3    "//tensorflow/core/platform:build_config_root.bzl",
4    "tf_gpu_tests_tags",
5)
6
7package(
8    default_visibility = ["//visibility:public"],
9    licenses = ["notice"],  # Apache 2.0
10)
11
12cc_library(
13    name = "converter",
14    srcs = ["converter.cc"],
15    hdrs = ["converter.h"],
16    deps = [
17        "//tensorflow/lite/delegates/gpu:spi",
18        "//tensorflow/lite/delegates/gpu/common:shape",
19        "//tensorflow/lite/delegates/gpu/common:status",
20        "//tensorflow/lite/delegates/gpu/common:types",
21        "//tensorflow/lite/delegates/gpu/common:util",
22        "//tensorflow/lite/delegates/gpu/gl:command_queue",
23        "//tensorflow/lite/delegates/gpu/gl:gl_buffer",
24        "//tensorflow/lite/delegates/gpu/gl:gl_program",
25        "//tensorflow/lite/delegates/gpu/gl:gl_shader",
26        "@com_google_absl//absl/strings",
27        "@com_google_absl//absl/types:span",
28    ],
29)
30
31cc_test(
32    name = "converter_test",
33    size = "small",
34    srcs = ["converter_test.cc"],
35    linkopts = [
36        "-lEGL",
37        "-lGLESv3",
38    ],
39    tags = tf_gpu_tests_tags() + [
40        "local",
41        "nobuilder",
42        "notap",
43        "tflite_not_portable_ios",
44    ],
45    deps = [
46        ":converter",
47        "//tensorflow/lite/delegates/gpu/common:convert",
48        "//tensorflow/lite/delegates/gpu/common:shape",
49        "//tensorflow/lite/delegates/gpu/common:status",
50        "//tensorflow/lite/delegates/gpu/gl:egl_environment",
51        "//tensorflow/lite/delegates/gpu/gl:gl_buffer",
52        "//tensorflow/lite/delegates/gpu/gl:portable",
53        "@com_google_absl//absl/types:span",
54        "@com_google_googletest//:gtest_main",
55    ],
56)
57
58cc_library(
59    name = "add",
60    srcs = ["add.cc"],
61    hdrs = ["add.h"],
62    deps = [
63        "//tensorflow/lite/delegates/gpu/common:data_type",
64        "//tensorflow/lite/delegates/gpu/common:operations",
65        "//tensorflow/lite/delegates/gpu/common:status",
66        "//tensorflow/lite/delegates/gpu/common:types",
67        "//tensorflow/lite/delegates/gpu/gl:node_shader",
68        "@com_google_absl//absl/memory",
69        "@com_google_absl//absl/strings",
70    ],
71)
72
73cc_test(
74    name = "add_test",
75    srcs = ["add_test.cc"],
76    linkstatic = True,
77    tags = tf_gpu_tests_tags() + [
78        "notap",
79        "tflite_not_portable_ios",
80    ],
81    deps = [
82        ":add",
83        ":test_util",
84        "//tensorflow/lite/delegates/gpu/common:operations",
85        "@com_google_googletest//:gtest",
86    ],
87)
88
89cc_library(
90    name = "concat",
91    srcs = ["concat.cc"],
92    hdrs = ["concat.h"],
93    deps = [
94        "//tensorflow/lite/delegates/gpu/common:operations",
95        "//tensorflow/lite/delegates/gpu/common:status",
96        "//tensorflow/lite/delegates/gpu/common:types",
97        "//tensorflow/lite/delegates/gpu/gl:node_shader",
98        "//tensorflow/lite/delegates/gpu/gl:variable",
99        "@com_google_absl//absl/memory",
100    ],
101)
102
103cc_test(
104    name = "concat_test",
105    srcs = ["concat_test.cc"],
106    linkstatic = True,
107    tags = tf_gpu_tests_tags() + [
108        "notap",
109        "tflite_not_portable_ios",
110    ],
111    deps = [
112        ":concat",
113        ":test_util",
114        "//tensorflow/lite/delegates/gpu/common:operations",
115        "@com_google_googletest//:gtest",
116    ],
117)
118
119cc_library(
120    name = "conv",
121    srcs = ["conv.cc"],
122    hdrs = ["conv.h"],
123    deps = [
124        "//tensorflow/lite/delegates/gpu/common:convert",
125        "//tensorflow/lite/delegates/gpu/common:operations",
126        "//tensorflow/lite/delegates/gpu/common:shape",
127        "//tensorflow/lite/delegates/gpu/common:status",
128        "//tensorflow/lite/delegates/gpu/common:types",
129        "//tensorflow/lite/delegates/gpu/common:util",
130        "//tensorflow/lite/delegates/gpu/gl:node_shader",
131        "//tensorflow/lite/delegates/gpu/gl:variable",
132        "//tensorflow/lite/delegates/gpu/gl/workgroups:ideal_workgroup_picker",
133        "@com_google_absl//absl/memory",
134        "@com_google_absl//absl/strings",
135    ],
136)
137
138cc_test(
139    name = "conv_test",
140    srcs = ["conv_test.cc"],
141    linkstatic = True,
142    tags = tf_gpu_tests_tags() + [
143        "notap",
144        "tflite_not_portable_ios",
145    ],
146    deps = [
147        ":conv",
148        ":test_util",
149        "//tensorflow/lite/delegates/gpu/common:operations",
150        "@com_google_googletest//:gtest",
151    ],
152)
153
154cc_library(
155    name = "custom_registry",
156    srcs = ["custom_registry.cc"],
157    hdrs = ["custom_registry.h"],
158    deps = [
159        "//tensorflow/lite/delegates/gpu/gl:node_shader",
160        "@com_google_absl//absl/container:flat_hash_map",
161    ],
162)
163
164cc_library(
165    name = "depthwise_conv",
166    srcs = ["depthwise_conv.cc"],
167    hdrs = ["depthwise_conv.h"],
168    deps = [
169        "//tensorflow/lite/delegates/gpu/common:convert",
170        "//tensorflow/lite/delegates/gpu/common:operations",
171        "//tensorflow/lite/delegates/gpu/common:shape",
172        "//tensorflow/lite/delegates/gpu/common:status",
173        "//tensorflow/lite/delegates/gpu/common:types",
174        "//tensorflow/lite/delegates/gpu/common:util",
175        "//tensorflow/lite/delegates/gpu/gl:node_shader",
176        "//tensorflow/lite/delegates/gpu/gl:variable",
177        "//tensorflow/lite/delegates/gpu/gl/workgroups:ideal_workgroup_picker",
178        "@com_google_absl//absl/memory",
179    ],
180)
181
182cc_test(
183    name = "depthwise_conv_test",
184    srcs = ["depthwise_conv_test.cc"],
185    linkstatic = True,
186    tags = tf_gpu_tests_tags() + [
187        "notap",
188        "tflite_not_portable_ios",
189    ],
190    deps = [
191        ":depthwise_conv",
192        ":test_util",
193        "//tensorflow/lite/delegates/gpu/common:operations",
194        "@com_google_googletest//:gtest",
195    ],
196)
197
198cc_library(
199    name = "elementwise",
200    srcs = ["elementwise.cc"],
201    hdrs = ["elementwise.h"],
202    deps = [
203        "//tensorflow/lite/delegates/gpu/common:operations",
204        "//tensorflow/lite/delegates/gpu/common:status",
205        "//tensorflow/lite/delegates/gpu/common:types",
206        "//tensorflow/lite/delegates/gpu/gl:node_shader",
207        "@com_google_absl//absl/memory",
208        "@com_google_absl//absl/strings",
209    ],
210)
211
212cc_test(
213    name = "elementwise_test",
214    srcs = ["elementwise_test.cc"],
215    linkstatic = True,
216    tags = tf_gpu_tests_tags() + [
217        "notap",
218        "tflite_not_portable_ios",
219    ],
220    deps = [
221        ":elementwise",
222        ":test_util",
223        "//tensorflow/lite/delegates/gpu/common:operations",
224        "@com_google_googletest//:gtest",
225    ],
226)
227
228cc_library(
229    name = "fully_connected",
230    srcs = ["fully_connected.cc"],
231    hdrs = ["fully_connected.h"],
232    deps = [
233        "//tensorflow/lite/delegates/gpu/common:convert",
234        "//tensorflow/lite/delegates/gpu/common:operations",
235        "//tensorflow/lite/delegates/gpu/common:status",
236        "//tensorflow/lite/delegates/gpu/common:types",
237        "//tensorflow/lite/delegates/gpu/gl:node_shader",
238        "//tensorflow/lite/delegates/gpu/gl:variable",
239        "@com_google_absl//absl/memory",
240    ],
241)
242
243cc_test(
244    name = "fully_connected_test",
245    srcs = ["fully_connected_test.cc"],
246    linkstatic = True,
247    tags = tf_gpu_tests_tags() + [
248        "notap",
249        "tflite_not_portable_ios",
250    ],
251    deps = [
252        ":fully_connected",
253        ":test_util",
254        "//tensorflow/lite/delegates/gpu/common:operations",
255        "@com_google_googletest//:gtest",
256    ],
257)
258
259cc_library(
260    name = "lstm",
261    srcs = ["lstm.cc"],
262    hdrs = ["lstm.h"],
263    deps = [
264        "//tensorflow/lite/delegates/gpu/common:operations",
265        "//tensorflow/lite/delegates/gpu/common:status",
266        "//tensorflow/lite/delegates/gpu/common:types",
267        "//tensorflow/lite/delegates/gpu/gl:node_shader",
268        "@com_google_absl//absl/memory",
269    ],
270)
271
272cc_test(
273    name = "lstm_test",
274    srcs = ["lstm_test.cc"],
275    linkstatic = True,
276    tags = tf_gpu_tests_tags() + [
277        "notap",
278        "tflite_not_portable_ios",
279    ],
280    deps = [
281        ":lstm",
282        ":test_util",
283        "//tensorflow/lite/delegates/gpu/common:operations",
284        "@com_google_googletest//:gtest",
285    ],
286)
287
288cc_library(
289    name = "max_unpooling",
290    srcs = ["max_unpooling.cc"],
291    hdrs = ["max_unpooling.h"],
292    deps = [
293        "//tensorflow/lite/delegates/gpu/common:operations",
294        "//tensorflow/lite/delegates/gpu/common:status",
295        "//tensorflow/lite/delegates/gpu/common:types",
296        "//tensorflow/lite/delegates/gpu/gl:node_shader",
297        "//tensorflow/lite/delegates/gpu/gl:variable",
298        "@com_google_absl//absl/memory",
299    ],
300)
301
302cc_test(
303    name = "max_unpooling_test",
304    srcs = ["max_unpooling_test.cc"],
305    linkstatic = True,
306    tags = tf_gpu_tests_tags() + [
307        "notap",
308        "tflite_not_portable_ios",
309    ],
310    deps = [
311        ":max_unpooling",
312        ":test_util",
313        "//tensorflow/lite/delegates/gpu/common:operations",
314        "@com_google_googletest//:gtest",
315    ],
316)
317
318cc_library(
319    name = "mean",
320    srcs = ["mean.cc"],
321    hdrs = ["mean.h"],
322    deps = [
323        "//tensorflow/lite/delegates/gpu/common:operations",
324        "//tensorflow/lite/delegates/gpu/common:status",
325        "//tensorflow/lite/delegates/gpu/common:types",
326        "//tensorflow/lite/delegates/gpu/common:util",
327        "//tensorflow/lite/delegates/gpu/gl:node_shader",
328        "@com_google_absl//absl/memory",
329        "@com_google_absl//absl/status",
330    ],
331)
332
333cc_test(
334    name = "mean_test",
335    srcs = ["mean_test.cc"],
336    linkstatic = True,
337    tags = [
338        "no_mac",  # TODO(b/171882090)
339        "notap",
340        "tflite_not_portable_ios",
341    ],
342    deps = [
343        ":mean",
344        ":test_util",
345        "//tensorflow/lite/delegates/gpu/common:operations",
346        "@com_google_googletest//:gtest",
347    ],
348)
349
350cc_library(
351    name = "mul",
352    srcs = ["mul.cc"],
353    hdrs = ["mul.h"],
354    deps = [
355        "//tensorflow/lite/delegates/gpu/common:convert",
356        "//tensorflow/lite/delegates/gpu/common:operations",
357        "//tensorflow/lite/delegates/gpu/common:status",
358        "//tensorflow/lite/delegates/gpu/common:types",
359        "//tensorflow/lite/delegates/gpu/gl:node_shader",
360        "@com_google_absl//absl/memory",
361        "@com_google_absl//absl/strings",
362    ],
363)
364
365cc_test(
366    name = "mul_test",
367    srcs = ["mul_test.cc"],
368    linkstatic = True,
369    tags = tf_gpu_tests_tags() + [
370        "notap",
371        "tflite_not_portable_ios",
372    ],
373    deps = [
374        ":mul",
375        ":test_util",
376        "//tensorflow/lite/delegates/gpu/common:operations",
377        "@com_google_googletest//:gtest",
378    ],
379)
380
381cc_library(
382    name = "pad",
383    srcs = ["pad.cc"],
384    hdrs = ["pad.h"],
385    deps = [
386        "//tensorflow/lite/delegates/gpu/common:operations",
387        "//tensorflow/lite/delegates/gpu/common:status",
388        "//tensorflow/lite/delegates/gpu/common:types",
389        "//tensorflow/lite/delegates/gpu/gl:node_shader",
390        "//tensorflow/lite/delegates/gpu/gl:variable",
391        "@com_google_absl//absl/memory",
392    ],
393)
394
395cc_test(
396    name = "pad_test",
397    srcs = ["pad_test.cc"],
398    linkstatic = True,
399    tags = tf_gpu_tests_tags() + [
400        "notap",
401        "tflite_not_portable_ios",
402    ],
403    deps = [
404        ":pad",
405        ":test_util",
406        "//tensorflow/lite/delegates/gpu/common:operations",
407        "@com_google_googletest//:gtest",
408    ],
409)
410
411cc_library(
412    name = "pooling",
413    srcs = ["pooling.cc"],
414    hdrs = ["pooling.h"],
415    deps = [
416        "//tensorflow/lite/delegates/gpu/common:operations",
417        "//tensorflow/lite/delegates/gpu/common:status",
418        "//tensorflow/lite/delegates/gpu/common:types",
419        "//tensorflow/lite/delegates/gpu/gl:node_shader",
420        "//tensorflow/lite/delegates/gpu/gl:variable",
421        "@com_google_absl//absl/memory",
422    ],
423)
424
425cc_test(
426    name = "pooling_test",
427    srcs = ["pooling_test.cc"],
428    linkstatic = True,
429    tags = tf_gpu_tests_tags() + [
430        "notap",
431        "tflite_not_portable_ios",
432    ],
433    deps = [
434        ":pooling",
435        ":test_util",
436        "//tensorflow/lite/delegates/gpu/common:operations",
437        "@com_google_googletest//:gtest",
438    ],
439)
440
441cc_library(
442    name = "prelu",
443    srcs = ["prelu.cc"],
444    hdrs = ["prelu.h"],
445    deps = [
446        "//tensorflow/lite/delegates/gpu/common:convert",
447        "//tensorflow/lite/delegates/gpu/common:data_type",
448        "//tensorflow/lite/delegates/gpu/common:operations",
449        "//tensorflow/lite/delegates/gpu/common:shape",
450        "//tensorflow/lite/delegates/gpu/common:status",
451        "//tensorflow/lite/delegates/gpu/common:types",
452        "//tensorflow/lite/delegates/gpu/gl:node_shader",
453        "@com_google_absl//absl/memory",
454    ],
455)
456
457cc_test(
458    name = "prelu_test",
459    srcs = ["prelu_test.cc"],
460    linkstatic = True,
461    tags = tf_gpu_tests_tags() + [
462        "notap",
463        "tflite_not_portable_ios",
464    ],
465    deps = [
466        ":prelu",
467        ":test_util",
468        "//tensorflow/lite/delegates/gpu/common:operations",
469        "@com_google_googletest//:gtest",
470    ],
471)
472
473cc_library(
474    name = "quantize_and_dequantize",
475    srcs = ["quantize_and_dequantize.cc"],
476    hdrs = ["quantize_and_dequantize.h"],
477    deps = [
478        "//tensorflow/lite/delegates/gpu/common:data_type",
479        "//tensorflow/lite/delegates/gpu/common:operations",
480        "//tensorflow/lite/delegates/gpu/common:shape",
481        "//tensorflow/lite/delegates/gpu/common:status",
482        "//tensorflow/lite/delegates/gpu/common:types",
483        "//tensorflow/lite/delegates/gpu/gl:node_shader",
484        "@com_google_absl//absl/memory",
485    ],
486)
487
488cc_test(
489    name = "quantize_and_dequantize_test",
490    srcs = ["quantize_and_dequantize_test.cc"],
491    linkstatic = True,
492    tags = tf_gpu_tests_tags() + [
493        "notap",
494        "tflite_not_portable_ios",
495    ],
496    deps = [
497        ":quantize_and_dequantize",
498        ":test_util",
499        "//tensorflow/lite/delegates/gpu/common:operations",
500        "//tensorflow/lite/kernels/internal:quantization_util",
501        "@com_google_googletest//:gtest",
502    ],
503)
504
505cc_library(
506    name = "relu",
507    srcs = ["relu.cc"],
508    hdrs = ["relu.h"],
509    deps = [
510        "//tensorflow/lite/delegates/gpu/common:operations",
511        "//tensorflow/lite/delegates/gpu/common:status",
512        "//tensorflow/lite/delegates/gpu/common:types",
513        "//tensorflow/lite/delegates/gpu/gl:node_shader",
514        "//tensorflow/lite/delegates/gpu/gl:variable",
515        "@com_google_absl//absl/memory",
516    ],
517)
518
519cc_test(
520    name = "relu_test",
521    srcs = ["relu_test.cc"],
522    linkstatic = True,
523    tags = tf_gpu_tests_tags() + [
524        "notap",
525        "tflite_not_portable_ios",
526    ],
527    deps = [
528        ":relu",
529        ":test_util",
530        "//tensorflow/lite/delegates/gpu/common:operations",
531        "@com_google_googletest//:gtest",
532    ],
533)
534
535cc_library(
536    name = "reshape",
537    srcs = ["reshape.cc"],
538    hdrs = ["reshape.h"],
539    deps = [
540        "//tensorflow/lite/delegates/gpu/common:operations",
541        "//tensorflow/lite/delegates/gpu/common:status",
542        "//tensorflow/lite/delegates/gpu/common:types",
543        "//tensorflow/lite/delegates/gpu/gl:node_shader",
544        "@com_google_absl//absl/memory",
545    ],
546)
547
548cc_test(
549    name = "reshape_test",
550    srcs = ["reshape_test.cc"],
551    linkstatic = True,
552    tags = tf_gpu_tests_tags() + [
553        "notap",
554        "tflite_not_portable_ios",
555    ],
556    deps = [
557        ":reshape",
558        ":test_util",
559        "//tensorflow/lite/delegates/gpu/common:operations",
560        "@com_google_googletest//:gtest",
561    ],
562)
563
564cc_library(
565    name = "slice",
566    srcs = ["slice.cc"],
567    hdrs = ["slice.h"],
568    deps = [
569        "//tensorflow/lite/delegates/gpu/common:operations",
570        "//tensorflow/lite/delegates/gpu/common:status",
571        "//tensorflow/lite/delegates/gpu/common:types",
572        "//tensorflow/lite/delegates/gpu/gl:node_shader",
573        "//tensorflow/lite/delegates/gpu/gl:variable",
574        "@com_google_absl//absl/memory",
575    ],
576)
577
578cc_test(
579    name = "slice_test",
580    srcs = ["slice_test.cc"],
581    linkstatic = True,
582    tags = tf_gpu_tests_tags() + [
583        "notap",
584        "tflite_not_portable_ios",
585    ],
586    deps = [
587        ":slice",
588        ":test_util",
589        "//tensorflow/lite/delegates/gpu/common:operations",
590        "@com_google_googletest//:gtest",
591    ],
592)
593
594cc_library(
595    name = "softmax",
596    srcs = ["softmax.cc"],
597    hdrs = ["softmax.h"],
598    deps = [
599        "//tensorflow/lite/delegates/gpu/common:operations",
600        "//tensorflow/lite/delegates/gpu/common:shape",
601        "//tensorflow/lite/delegates/gpu/common:status",
602        "//tensorflow/lite/delegates/gpu/common:types",
603        "//tensorflow/lite/delegates/gpu/common:util",
604        "//tensorflow/lite/delegates/gpu/gl:node_shader",
605        "//tensorflow/lite/delegates/gpu/gl:variable",
606        "@com_google_absl//absl/memory",
607    ],
608)
609
610cc_test(
611    name = "softmax_test",
612    srcs = ["softmax_test.cc"],
613    linkstatic = True,
614    tags = tf_gpu_tests_tags() + [
615        "notap",
616        "tflite_not_portable_ios",
617    ],
618    deps = [
619        ":softmax",
620        ":test_util",
621        "//tensorflow/lite/delegates/gpu/common:operations",
622        "//tensorflow/lite/delegates/gpu/common:shape",
623        "@com_google_googletest//:gtest",
624    ],
625)
626
627cc_library(
628    name = "space_to_depth",
629    srcs = ["space_to_depth.cc"],
630    hdrs = ["space_to_depth.h"],
631    deps = [
632        "//tensorflow/lite/delegates/gpu/common:operations",
633        "//tensorflow/lite/delegates/gpu/common:status",
634        "//tensorflow/lite/delegates/gpu/gl:node_shader",
635        "@com_google_absl//absl/memory",
636        "@com_google_absl//absl/types:any",
637    ],
638)
639
640cc_test(
641    name = "space_to_depth_test",
642    srcs = ["space_to_depth_test.cc"],
643    linkstatic = True,
644    tags = tf_gpu_tests_tags() + [
645        "notap",
646        "tflite_not_portable_ios",
647    ],
648    deps = [
649        ":space_to_depth",
650        ":test_util",
651        "//tensorflow/lite/delegates/gpu/common:operations",
652        "@com_google_googletest//:gtest",
653    ],
654)
655
656cc_library(
657    name = "test_util",
658    testonly = 1,
659    srcs = ["test_util.cc"],
660    hdrs = ["test_util.h"],
661    linkopts = [
662        "-lEGL",
663        "-lGLESv3",
664    ],
665    deps = [
666        "//tensorflow/lite/delegates/gpu/common:model",
667        "//tensorflow/lite/delegates/gpu/common:operations",
668        "//tensorflow/lite/delegates/gpu/common:status",
669        "//tensorflow/lite/delegates/gpu/common:tensor",
670        "//tensorflow/lite/delegates/gpu/gl:api",
671        "//tensorflow/lite/delegates/gpu/gl:compiler_options",
672        "//tensorflow/lite/delegates/gpu/gl:egl_environment",
673        "//tensorflow/lite/delegates/gpu/gl:gl_buffer",
674        "//tensorflow/lite/delegates/gpu/gl:node_shader",
675        "//tensorflow/lite/delegates/gpu/gl:object_manager",
676        "//tensorflow/lite/delegates/gpu/gl:request_gpu_info",
677        "//tensorflow/lite/delegates/gpu/gl:runtime_options",
678        "//tensorflow/lite/delegates/gpu/gl/workgroups:default_calculator",
679        "@com_google_absl//absl/container:flat_hash_map",
680        "@com_google_absl//absl/container:flat_hash_set",
681        "@com_google_googletest//:gtest",
682        "@com_google_googletest//:gtest_main",
683    ],
684)
685
686cc_library(
687    name = "transpose_conv",
688    srcs = ["transpose_conv.cc"],
689    hdrs = ["transpose_conv.h"],
690    deps = [
691        "//tensorflow/lite/delegates/gpu/common:convert",
692        "//tensorflow/lite/delegates/gpu/common:operations",
693        "//tensorflow/lite/delegates/gpu/common:shape",
694        "//tensorflow/lite/delegates/gpu/common:status",
695        "//tensorflow/lite/delegates/gpu/common:types",
696        "//tensorflow/lite/delegates/gpu/common:util",
697        "//tensorflow/lite/delegates/gpu/gl:node_shader",
698        "//tensorflow/lite/delegates/gpu/gl:variable",
699        "@com_google_absl//absl/memory",
700    ],
701)
702
703cc_test(
704    name = "transpose_conv_test",
705    srcs = ["transpose_conv_test.cc"],
706    linkstatic = True,
707    tags = tf_gpu_tests_tags() + [
708        "notap",
709        "tflite_not_portable_ios",
710    ],
711    deps = [
712        ":test_util",
713        ":transpose_conv",
714        "//tensorflow/lite/delegates/gpu/common:operations",
715        "@com_google_googletest//:gtest",
716    ],
717)
718
719cc_library(
720    name = "resize",
721    srcs = ["resize.cc"],
722    hdrs = ["resize.h"],
723    deps = [
724        "//tensorflow/lite/delegates/gpu/common:operations",
725        "//tensorflow/lite/delegates/gpu/common:status",
726        "//tensorflow/lite/delegates/gpu/common:types",
727        "//tensorflow/lite/delegates/gpu/gl:node_shader",
728        "@com_google_absl//absl/memory",
729    ],
730)
731
732cc_test(
733    name = "resize_test",
734    srcs = ["resize_test.cc"],
735    linkstatic = True,
736    tags = tf_gpu_tests_tags() + [
737        "notap",
738        "tflite_not_portable_ios",
739    ],
740    deps = [
741        ":resize",
742        ":test_util",
743        "//tensorflow/lite/delegates/gpu/common:operations",
744        "@com_google_googletest//:gtest",
745    ],
746)
747
748TFLITE_GPU_BINARY_RELEASE_OPERATORS = [
749    "add",
750    "concat",
751    "conv",
752    "depthwise_conv",
753    "elementwise",
754    "fully_connected",
755    "lstm",
756    "mul",
757    "pad",
758    "pooling",
759    "prelu",
760    "quantize_and_dequantize",
761    "relu",
762    "mean",
763    "reshape",
764    "resize",
765    "slice",
766    "softmax",
767    "space_to_depth",
768    "transpose_conv",
769]
770
771NON_TFLITE_GPU_BINARY_RELEASE_OPERATORS = [
772    "max_unpooling",
773]
774
775cc_library(
776    name = "registry",
777    srcs = ["registry.cc"],
778    hdrs = ["registry.h"],
779    visibility = ["//visibility:public"],
780    deps = [":" + op_name for op_name in TFLITE_GPU_BINARY_RELEASE_OPERATORS] +
781           select({
782               "//tensorflow/lite/delegates/gpu:tflite_gpu_binary_release": [],
783               "//conditions:default": NON_TFLITE_GPU_BINARY_RELEASE_OPERATORS,
784           }) + [
785        ":custom_registry",
786        "@com_google_absl//absl/container:flat_hash_map",
787        "//tensorflow/lite/delegates/gpu/common:operations",
788        "//tensorflow/lite/delegates/gpu/common:status",
789        "//tensorflow/lite/delegates/gpu/gl:node_shader",
790        "@com_google_absl//absl/memory",
791        "@com_google_absl//absl/strings",
792    ],
793)
794
795tflite_portable_test_suite_combined(combine_conditions = {"deps": [":test_util"]})
796