Lines Matching full:tensorflow
1 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
4 load("//tensorflow:tensorflow.bzl", "cc_header_only_library", "tf_cc_test")
7 load("//tensorflow:tensorflow.bzl", "if_libtpu", "if_with_tpu_support", "tf_copts")
8 load("//tensorflow/stream_executor:build_defs.bzl", "if_cuda_or_rocm")
11 load("//tensorflow:tensorflow.bzl", "cc_header_only_library")
14 load("//tensorflow:tensorflow.bzl", "filegroup")
17 load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library", "tf_jit_compilation_passes_extra_dep…
18 load("//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library")
20 "//tensorflow/core/platform:build_config_root.bzl",
36 "//tensorflow/compiler/tf2xla:internal",
39 "//tensorflow/c/...",
40 "//tensorflow/compiler/tests/...",
41 "//tensorflow/python/...",
48 "//tensorflow/compiler/tf2xla:friends",
73 "//tensorflow/compiler/plugin",
90 "//tensorflow/compiler/jit/kernels:xla_ops",
91 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
92 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
94 if_false = ["//tensorflow/compiler/xla/service:cpu_plugin"],
106 "//tensorflow/compiler/jit/kernels:xla_ops",
107 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
108 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
109 "//tensorflow/compiler/xla/service:gpu_plugin",
118 "//tensorflow/core/tpu/graph_rewrite:tpu_rewrite_pass_registration",
119 "//tensorflow/core/tpu/graph_rewrite:configure_tpu_embedding_rewrite_registration",
120 "//tensorflow/stream_executor/tpu:tpu_transfer_manager",
136 "//tensorflow/compiler/jit/kernels:xla_ops",
137 "//tensorflow/compiler/tf2xla:layout_util",
138 "//tensorflow/compiler/tf2xla:xla_compiler",
139 "//tensorflow/compiler/tf2xla:xla_op_registry",
140 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
141 "//tensorflow/core:core_cpu_internal",
142 "//tensorflow/core:lib",
145 "//tensorflow/compiler/xla/service:cpu_plugin", # buildcleaner: keep
165 "//tensorflow/compiler/jit/kernels:xla_ops",
166 "//tensorflow/compiler/tf2xla:layout_util",
167 "//tensorflow/compiler/tf2xla:xla_compiler",
168 "//tensorflow/compiler/tf2xla:xla_op_registry",
169 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
170 "//tensorflow/core:core_cpu_internal",
171 "//tensorflow/core:lib",
172 "//tensorflow/core/common_runtime/gpu:gpu_init",
175 "//tensorflow/compiler/xla/service:gpu_plugin", # buildcleaner: keep
191 "//tensorflow/compiler/jit/kernels:xla_ops",
192 "//tensorflow/compiler/tf2xla:common",
193 "//tensorflow/compiler/tf2xla:layout_util",
194 "//tensorflow/compiler/tf2xla:tf2xla_util",
195 "//tensorflow/compiler/tf2xla:xla_helpers",
196 "//tensorflow/compiler/tf2xla:xla_op_registry",
197 "//tensorflow/core:framework_internal",
198 "//tensorflow/core:lib_proto_parsing",
199 "//tensorflow/core:protos_all_cc",
200 "//tensorflow/core:session_options",
201 "//tensorflow/core/common_runtime:device",
202 "//tensorflow/core/common_runtime:device_factory",
203 "//tensorflow/core/common_runtime:dma_helper",
204 "//tensorflow/core/platform:status",
205 "//tensorflow/core/tpu:tpu_api",
206 "//tensorflow/core/tpu:tpu_defs",
207 "//tensorflow/core/tpu:tpu_node_device_util",
208 "//tensorflow/core/tpu:virtual_device",
209 "//tensorflow/stream_executor/tpu:c_api_conversions",
210 "//tensorflow/stream_executor/tpu:status_helper",
211 "//tensorflow/stream_executor/tpu:tpu_executor_base",
212 "//tensorflow/stream_executor/tpu:tpu_node_context",
213 "//tensorflow/stream_executor/tpu:tpu_platform_interface",
214 "//tensorflow/stream_executor/tpu:tpu_stream_interface",
216 "//tensorflow/core/common_runtime:copy_tensor",
228 "//tensorflow/compiler/tf2xla:common",
229 "//tensorflow/compiler/xla:shape_util",
230 "//tensorflow/compiler/xla/client:local_client",
231 "//tensorflow/compiler/xla/service:shaped_buffer",
232 "//tensorflow/core:core_cpu_internal",
233 "//tensorflow/core:framework",
234 "//tensorflow/core:lib",
235 "//tensorflow/core:lib_internal",
249 "//tensorflow/compiler/jit/ops:xla_ops",
250 "//tensorflow/compiler/tf2xla:layout_util",
251 "//tensorflow/compiler/tf2xla:common",
252 "//tensorflow/compiler/tf2xla:tf2xla_util",
253 "//tensorflow/compiler/tf2xla:xla_compiler",
254 "//tensorflow/compiler/tf2xla:xla_op_registry",
255 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
256 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
257 "//tensorflow/compiler/xla:util",
258 "//tensorflow/compiler/xla/client:client_library",
259 "//tensorflow/compiler/xla/client:global_data",
260 "//tensorflow/compiler/xla/client:local_client",
261 "//tensorflow/compiler/xla/service:stream_pool",
262 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options",
263 "//tensorflow/core:array_ops_op_lib",
264 "//tensorflow/core:control_flow_ops_op_lib",
265 "//tensorflow/core:core_cpu",
266 "//tensorflow/core:core_cpu_internal",
267 "//tensorflow/core:dataset_ops_op_lib",
268 "//tensorflow/core:framework",
269 "//tensorflow/core:framework_internal",
270 "//tensorflow/core:functional_ops_op_lib",
271 "//tensorflow/core:lib",
272 "//tensorflow/core:lib_internal",
273 "//tensorflow/core:math_ops_op_lib",
274 "//tensorflow/core:nn_ops_op_lib",
275 "//tensorflow/core:no_op_op_lib",
276 "//tensorflow/core:protos_all_cc",
277 "//tensorflow/core:resource_variable_ops_op_lib",
278 "//tensorflow/core:sendrecv_ops_op_lib",
279 "//tensorflow/core:state_ops_op_lib",
280 "//tensorflow/core/platform:stream_executor_no_cuda",
281 "//tensorflow/core/kernels:constant_op",
282 "//tensorflow/core/kernels:fifo_queue",
283 "//tensorflow/core/kernels:function_ops",
284 "//tensorflow/core/kernels:identity_op",
285 "//tensorflow/core/kernels:resource_variable_ops",
286 "//tensorflow/core/kernels:shape_ops",
287 "//tensorflow/core/kernels:variable_ops",
288 "//tensorflow/core/kernels/data:finalize_dataset_op",
289 "//tensorflow/core/kernels/data:generator_dataset_op",
290 "//tensorflow/core/kernels/data:iterator_ops",
291 "//tensorflow/core/kernels/data:optional_ops",
292 "//tensorflow/core/kernels/data:prefetch_dataset_op",
293 "//tensorflow/core/kernels/data:options_dataset_op",
294 "//tensorflow/core/profiler/lib:traceme",
295 "//tensorflow/stream_executor:tf_allocator_adapter",
296 "//tensorflow/stream_executor/platform",
347 "//tensorflow:android": [
348 "//tensorflow/core:portable_tensorflow_lib",
351 "//tensorflow/core:graph",
362 "//tensorflow/compiler/mlir/tensorflow:dump_graph",
363 "//tensorflow/compiler/xla:parse_flags_from_env",
364 "//tensorflow/core:framework_internal",
365 "//tensorflow/core:lib",
366 "//tensorflow/core:protos_all_cc",
380 "//tensorflow/compiler/mlir/tensorflow:dump_graph",
381 "//tensorflow/compiler/xla:parse_flags_from_env",
382 "//tensorflow/core:framework_internal",
383 "//tensorflow/core:lib",
384 "//tensorflow/core/protobuf:for_core_protos_cc",
425 "//tensorflow/compiler/tf2xla:common",
426 "//tensorflow/compiler/tf2xla:xla_compiler",
427 "//tensorflow/compiler/xla:shape_util",
428 "//tensorflow/compiler/xla:statusor",
429 "//tensorflow/compiler/xla/client:client_library",
430 "//tensorflow/compiler/xla/client:local_client",
431 "//tensorflow/compiler/xla/service:shaped_buffer",
432 "//tensorflow/core:core_cpu_internal",
433 "//tensorflow/core:framework",
434 "//tensorflow/core:framework_internal",
435 "//tensorflow/core:gpu_runtime",
436 "//tensorflow/core:lib",
437 "//tensorflow/core:lib_internal",
438 "//tensorflow/core:protos_all_cc",
439 "//tensorflow/stream_executor:device_memory_allocator",
450 protodeps = tf_additional_all_protos() + ["//tensorflow/compiler/xla/service:hlo_proto"],
465 "//tensorflow/compiler/mlir:array_container_utils",
466 "//tensorflow/compiler/mlir/tensorflow:compile_mlir_util_no_tf_dialect_passes",
467 "//tensorflow/compiler/mlir/tf2xla:mlir_bridge_rollout_policy",
468 "//tensorflow/compiler/tf2xla:common",
469 "//tensorflow/compiler/tf2xla:xla_compiler",
470 "//tensorflow/compiler/tf2xla:xla_context",
471 "//tensorflow/compiler/xla:protobuf_util",
472 "//tensorflow/compiler/xla:status_macros",
473 "//tensorflow/compiler/xla:statusor",
474 "//tensorflow/compiler/xla:util",
475 "//tensorflow/compiler/xla/client:client_library",
476 "//tensorflow/compiler/xla/client:local_client",
477 "//tensorflow/compiler/xla/service:compiler",
478 "//tensorflow/compiler/xla/service:hlo_proto_cc",
479 "//tensorflow/core:core_cpu",
480 "//tensorflow/core:core_cpu_internal",
481 "//tensorflow/core:framework",
482 "//tensorflow/core:lib",
483 "//tensorflow/core:lib_internal",
484 "//tensorflow/core:protos_all_cc",
485 "//tensorflow/core/platform:logging",
486 "//tensorflow/core/tpu:tpu_defs",
507 "//tensorflow/compiler/tf2xla:common",
508 "//tensorflow/compiler/xla/client:client_library",
509 "//tensorflow/core:test",
510 "//tensorflow/core:test_main",
523 "//tensorflow/compiler/tf2xla:common",
524 "//tensorflow/compiler/tf2xla:xla_compiler",
525 "//tensorflow/compiler/xla/client:client_library",
526 "//tensorflow/core:test",
527 "//tensorflow/core:test_main",
537 "//tensorflow/compiler/tf2xla:functionalize_control_flow_pass_registration",
538 "//tensorflow/compiler/tf2xla:mlir_bridge_pass_registration",
539 "//tensorflow/core:core_cpu_internal",
555 "//tensorflow/compiler/tf2xla:xla_compiler",
556 "//tensorflow/compiler/xla:statusor",
557 "//tensorflow/compiler/xla/client:executable_build_options",
558 "//tensorflow/compiler/xla/client:local_client",
559 "//tensorflow/compiler/xla/service:hlo_graph_dumper",
560 "//tensorflow/core:framework",
561 "//tensorflow/core:lib",
562 "//tensorflow/core/common_runtime:core_cpu_internal",
563 "//tensorflow/core/common_runtime/eager:tensor_handle",
579 "//tensorflow/compiler/xla:statusor",
615 "//tensorflow/core/common_runtime/eager:__pkg__",
623 "//tensorflow/compiler/jit/kernels:xla_ops_no_jit_rewrite_registration",
624 "//tensorflow/compiler/tf2xla:mlir_bridge_pass",
625 "//tensorflow/compiler/tf2xla:xla_compiler",
626 "//tensorflow/compiler/tf2xla:xla_op_registry",
627 "//tensorflow/core:core_cpu_internal",
628 "//tensorflow/core:framework",
629 "//tensorflow/core:lib",
630 "//tensorflow/core:protos_all_cc",
646 "//tensorflow/core:core_cpu_internal",
647 "//tensorflow/core:framework",
648 "//tensorflow/core:lib",
649 "//tensorflow/core:protos_all_cc",
650 "//tensorflow/core:session_options",
651 "//tensorflow/core:test",
652 "//tensorflow/core:test_main",
653 "//tensorflow/core:testlib",
664 "//tensorflow/compiler/tf2xla:resource_operation_table",
665 "//tensorflow/compiler/xla/service/graphcycles",
666 "//tensorflow/core:framework",
667 "//tensorflow/core:graph",
668 "//tensorflow/core:lib",
669 "//tensorflow/core:protos_all_cc",
683 "//tensorflow/cc:cc_ops",
684 "//tensorflow/cc:cc_ops_internal",
685 "//tensorflow/cc:function_ops",
686 "//tensorflow/cc:functional_ops",
687 "//tensorflow/cc:ops",
688 "//tensorflow/cc:resource_variable_ops",
689 "//tensorflow/cc:sendrecv_ops",
690 "//tensorflow/compiler/jit/kernels:xla_ops",
691 "//tensorflow/compiler/tf2xla:xla_compiler",
692 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
693 "//tensorflow/core:core_cpu",
694 "//tensorflow/core:framework",
695 "//tensorflow/core:framework_internal",
696 "//tensorflow/core:graph",
697 "//tensorflow/core:lib",
698 "//tensorflow/core:test",
699 "//tensorflow/core:test_main",
700 "//tensorflow/core:testlib",
712 "//tensorflow/compiler/xla:statusor",
713 "//tensorflow/core:core_cpu_internal",
714 "//tensorflow/core:framework",
715 "//tensorflow/core:graph",
716 "//tensorflow/core:lib",
717 "//tensorflow/core:protos_all_cc",
728 "//tensorflow/compiler/xla:status_macros",
729 "//tensorflow/core:core_cpu",
730 "//tensorflow/core:framework",
731 "//tensorflow/core:lib",
741 "//tensorflow/cc:cc_ops",
742 "//tensorflow/cc:cc_ops_internal",
743 "//tensorflow/cc:ops",
744 "//tensorflow/cc:resource_variable_ops",
745 "//tensorflow/core:framework",
746 "//tensorflow/core:ops",
747 "//tensorflow/core:test",
748 "//tensorflow/core:test_main",
749 "//tensorflow/core/kernels:constant_op",
759 "//tensorflow/compiler/tf2xla:tf2xla_util",
760 "//tensorflow/core:framework",
761 "//tensorflow/core:graph",
762 "//tensorflow/core:protos_all_cc",
763 "//tensorflow/stream_executor/lib",
776 "//tensorflow/cc:cc_ops",
777 "//tensorflow/cc:scope",
778 "//tensorflow/core:framework",
779 "//tensorflow/core:ops",
780 "//tensorflow/core:protos_all_cc",
781 "//tensorflow/core:test",
782 "//tensorflow/core:test_main",
823 "//tensorflow/core/tfrt/utils:__pkg__",
836 "//tensorflow/cc:cc_ops",
837 "//tensorflow/cc:functional_ops",
838 "//tensorflow/cc:ops",
839 "//tensorflow/cc:scope",
840 "//tensorflow/cc:scope_internal",
841 "//tensorflow/compiler/jit/ops:xla_ops",
842 "//tensorflow/compiler/tf2xla:resource_operation_table",
843 "//tensorflow/compiler/tf2xla:side_effect_util",
844 "//tensorflow/compiler/tf2xla:tf2xla_util",
845 "//tensorflow/compiler/tf2xla:xla_compiler",
846 "//tensorflow/compiler/tf2xla:xla_op_registry",
847 "//tensorflow/compiler/tf2xla/cc:xla_jit_ops",
848 "//tensorflow/compiler/tf2xla/cc:xla_ops",
849 "//tensorflow/compiler/xla:status_macros",
850 "//tensorflow/compiler/xla:statusor",
851 "//tensorflow/compiler/xla:union_find",
852 "//tensorflow/compiler/xla:util",
853 "//tensorflow/compiler/xla:xla_data_proto_cc",
854 "//tensorflow/compiler/xla/service/graphcycles",
855 "//tensorflow/core:core_cpu",
856 "//tensorflow/core:core_cpu_internal",
857 "//tensorflow/core:framework",
858 "//tensorflow/core:lib",
859 "//tensorflow/core:lib_internal",
860 "//tensorflow/core:protos_all_cc",
861 "//tensorflow/core/framework:bounds_check",
862 "//tensorflow/stream_executor/lib",
881 "//tensorflow/compiler/xla:status_macros",
882 "//tensorflow/compiler/xla:statusor",
883 "//tensorflow/compiler/xla/service/graphcycles",
884 "//tensorflow/core:core_cpu",
885 "//tensorflow/core:framework",
886 "//tensorflow/core:framework_internal",
887 "//tensorflow/core:graph",
888 "//tensorflow/core:lib",
889 "//tensorflow/core:protos_all_cc",
890 "//tensorflow/core/framework:bounds_check",
891 "//tensorflow/stream_executor/lib",
906 "//tensorflow/compiler/tf2xla:xla_compiler",
907 "//tensorflow/compiler/tf2xla:xla_op_registry",
908 "//tensorflow/compiler/xla:status_macros",
909 "//tensorflow/compiler/xla:statusor",
910 "//tensorflow/core:framework",
924 "//tensorflow/core:test",
925 "//tensorflow/core:test_main",
926 "//tensorflow/core:testlib",
940 "//tensorflow/cc:cc_ops",
941 "//tensorflow/cc:cc_ops_internal",
942 "//tensorflow/cc:function_ops",
943 "//tensorflow/cc:ops",
944 "//tensorflow/cc:sendrecv_ops",
945 "//tensorflow/compiler/jit/kernels:xla_ops",
946 "//tensorflow/compiler/tf2xla:xla_compiler",
947 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
948 "//tensorflow/core:core_cpu",
949 "//tensorflow/core:framework",
950 "//tensorflow/core:framework_internal",
951 "//tensorflow/core:graph",
952 "//tensorflow/core:lib",
953 "//tensorflow/core:test",
954 "//tensorflow/core:test_main",
955 "//tensorflow/core:testlib",
967 "//tensorflow/core:lib",
968 "//tensorflow/core:test",
1008 "//tensorflow/cc:cc_ops",
1009 "//tensorflow/cc:cc_ops_internal",
1010 "//tensorflow/cc:function_ops",
1011 "//tensorflow/cc:functional_ops",
1012 "//tensorflow/cc:ops",
1013 "//tensorflow/cc:resource_variable_ops",
1014 "//tensorflow/cc:scope",
1015 "//tensorflow/cc:sendrecv_ops",
1016 "//tensorflow/compiler/jit/kernels:xla_ops",
1017 "//tensorflow/compiler/tf2xla:rearrange_function_argument",
1018 "//tensorflow/compiler/tf2xla:side_effect_util",
1019 "//tensorflow/compiler/tf2xla:test_util",
1020 "//tensorflow/compiler/tf2xla:xla_compiler",
1021 "//tensorflow/compiler/tf2xla/cc:xla_jit_ops",
1022 "//tensorflow/compiler/tf2xla/cc:xla_ops",
1023 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
1024 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
1025 "//tensorflow/compiler/xla:test",
1026 "//tensorflow/core:all_kernels",
1027 "//tensorflow/core:core_cpu",
1028 "//tensorflow/core:framework",
1029 "//tensorflow/core:framework_internal",
1030 "//tensorflow/core:lib",
1031 "//tensorflow/core:protos_all_cc",
1032 "//tensorflow/core:session_options",
1033 "//tensorflow/core:test",
1034 "//tensorflow/core:testlib",
1051 "//tensorflow/cc:cc_ops",
1052 "//tensorflow/cc:cc_ops_internal",
1053 "//tensorflow/cc:function_ops",
1054 "//tensorflow/cc:functional_ops",
1055 "//tensorflow/cc:ops",
1056 "//tensorflow/compiler/jit/kernels:xla_ops",
1057 "//tensorflow/compiler/tf2xla:xla_compiler",
1058 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
1059 "//tensorflow/compiler/xla:status_macros",
1060 "//tensorflow/core:core_cpu",
1061 "//tensorflow/core:core_cpu_lib",
1062 "//tensorflow/core:framework",
1063 "//tensorflow/core:framework_internal",
1064 "//tensorflow/core:lib",
1065 "//tensorflow/core:test",
1066 "//tensorflow/core:test_main",
1067 "//tensorflow/core:testlib",
1079 "//tensorflow/cc:ops",
1080 "//tensorflow/compiler/xla:test",
1081 "//tensorflow/core:framework",
1082 "//tensorflow/core:graph",
1083 "//tensorflow/core:protos_all_cc",
1096 "//tensorflow/cc:cc_ops",
1097 "//tensorflow/cc:cc_ops_internal",
1098 "//tensorflow/cc:ops",
1099 "//tensorflow/core:ops",
1100 "//tensorflow/core:test_main",
1119 "//tensorflow/compiler/tf2xla:resource_operation_table",
1120 "//tensorflow/compiler/tf2xla:tf2xla_util",
1121 "//tensorflow/compiler/tf2xla:xla_compiler",
1122 "//tensorflow/compiler/tf2xla:xla_op_registry",
1123 "//tensorflow/compiler/xla:statusor",
1124 "//tensorflow/compiler/xla:union_find",
1125 "//tensorflow/compiler/xla:util",
1126 "//tensorflow/compiler/xla/service/graphcycles",
1127 "//tensorflow/core:core_cpu",
1128 "//tensorflow/core:framework",
1129 "//tensorflow/core:graph",
1130 "//tensorflow/core:lib",
1131 "//tensorflow/core:protos_all_cc",
1147 "//tensorflow/cc:cc_ops",
1148 "//tensorflow/cc:function_ops",
1149 "//tensorflow/cc:functional_ops",
1150 "//tensorflow/cc:ops",
1151 "//tensorflow/cc:scope",
1152 "//tensorflow/compiler/tf2xla:test_util",
1153 "//tensorflow/compiler/tf2xla:xla_compiler",
1154 "//tensorflow/compiler/tf2xla/cc:xla_jit_ops",
1155 "//tensorflow/compiler/tf2xla/cc:xla_ops",
1156 "//tensorflow/core:core_cpu",
1157 "//tensorflow/core:framework",
1158 "//tensorflow/core:ops",
1159 "//tensorflow/core:protos_all_cc",
1160 "//tensorflow/core:test",
1161 "//tensorflow/core:test_main",
1174 "//tensorflow/cc:cc_ops",
1175 "//tensorflow/cc:ops",
1176 "//tensorflow/core:all_kernels",
1177 "//tensorflow/core:core_cpu",
1178 "//tensorflow/core:framework",
1179 "//tensorflow/core:ops",
1180 "//tensorflow/core:test",
1181 "//tensorflow/core/common_runtime:direct_session_internal",
1182 "//tensorflow/core/kernels:cwise_op",
1183 "//tensorflow/core/kernels:matmul_op",
1184 "//tensorflow/core/kernels:partitioned_function_ops",
1190 kernels = ["//tensorflow/compiler/jit/ops:xla_ops"],
1195 "//tensorflow/compiler/jit/ops:xla_ops_grad",
1196 "//tensorflow/compiler/jit/ops:xla_ops_wrapper_py",
1207 "//tensorflow/core:lib",
1225 "//tensorflow/core:lib",