| # |
| # Public targets: |
| # ":platform" - Low-level and platform-specific Python code. |
| |
| # Placeholder: load py_proto_library |
| load("//tensorflow:pytype.default.bzl", "pytype_strict_library") |
| load("//tensorflow:strict.default.bzl", "py_strict_library") |
| load("//tensorflow:tensorflow.bzl", "VERSION", "cc_header_only_library", "clean_dep", "if_google", "if_mlir", "if_oss", "if_windows", "if_xla_available", "tf_enable_mlir_bridge", "tf_python_pybind_static_deps", "tsl_async_value_deps") |
| load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable", "pywrap_tensorflow_macro", "tf_external_workspace_visible", "tf_monitoring_python_deps", "tf_pybind_cc_library_wrapper", "tf_python_pybind_extension") |
| load( |
| "//tensorflow/core/platform:build_config.bzl", |
| "tf_additional_binary_deps", |
| "tf_proto_library", |
| ) |
| load( |
| "//tensorflow/core/platform:build_config_root.bzl", |
| "if_static", |
| "tf_additional_plugin_deps", |
| "tf_additional_profiler_deps", |
| ) |
| load( |
| "//tensorflow/tsl/platform/default:cuda_build_defs.bzl", |
| "if_cuda_is_configured", |
| ) |
| |
| # TODO(mdan): Break into per-directory files. |
| |
| visibility = [ |
| "//engedu/ml/tf_from_scratch:__pkg__", |
| "//third_party/cloud_tpu/convergence_tools:__subpackages__", |
| "//third_party/mlperf:__subpackages__", |
| "//tensorflow:internal", |
| "//tensorflow/dtensor:dtensor-internal", |
| "//tensorflow/lite/toco/python:__pkg__", |
| "//tensorflow_models:__subpackages__", |
| "//tensorflow_model_optimization:__subpackages__", |
| "//third_party/py/cleverhans:__subpackages__", |
| "//third_party/courier:__subpackages__", |
| "//third_party/py/courier:__subpackages__", |
| "//third_party/py/lingvo:__subpackages__", |
| "//third_party/py/reverb:__subpackages__", |
| "//third_party/py/tensorfn:__subpackages__", |
| "//third_party/py/neural_structured_learning:__subpackages__", |
| "//third_party/py/tensorflow_docs:__subpackages__", |
| "//third_party/py/tensorflow_examples:__subpackages__", |
| "//third_party/py/tensorflow_recommenders:__subpackages__", |
| "//third_party/py/tf_agents:__subpackages__", # For benchmarks. |
| "//third_party/py/tf_slim:__subpackages__", |
| "//third_party/py/keras:__subpackages__", |
| "//third_party/py/tf_keras:__subpackages__", |
| "//third_party/py/starcraft2:__subpackages__", |
| "//third_party/py/tensorflow_gnn:__subpackages__", |
| "//third_party/py/tensorflow_numerics:__subpackages__", |
| "//third_party/py/tensorflow_privacy:__subpackages__", |
| "//third_party/reverb:__subpackages__", |
| "//tensorflow_minigo:__subpackages__", |
| "//research/graph/fairness/inproc_fair_reg:__subpackages__", |
| ] |
| |
| package( |
| # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], |
| default_visibility = visibility, |
| licenses = ["notice"], |
| ) |
| |
| # Description: |
| |
| py_strict_library( |
| name = "python", |
| srcs = ["__init__.py"], |
| srcs_version = "PY3", |
| visibility = [ |
| "//tensorflow:__pkg__", |
| "//tensorflow/compiler/aot/tests:__pkg__", # TODO(b/34059704): remove when fixed |
| "//tensorflow/lite/toco/python:__pkg__", # TODO(b/34059704): remove when fixed |
| "//tensorflow/python/debug:__subpackages__", |
| "//tensorflow/python/tools:__pkg__", # TODO(b/34059704): remove when fixed |
| "//tensorflow/tools/quantization:__pkg__", # TODO(b/34059704): remove when fixed |
| ], |
| deps = [ |
| ":no_contrib", |
| "//tensorflow/python/estimator:estimator_py", |
| "//tensorflow/python/ops:gradient_checker_v2", |
| "//tensorflow/python/ops:stateful_random_ops", |
| "//tensorflow/python/ops/distributions", |
| "//tensorflow/python/ops/structured:structured_ops", |
| "//tensorflow/python/tpu:tpu_estimator", |
| "//tensorflow/python/tpu:tpu_noestimator", |
| ], |
| ) |
| |
| # Some of the dependencies in keras_lib are possibly outdated or not generally |
| # useful, so a future task is to determine a good set of dependencies. |
| # keras_lib may be removed in the future, it should NOT be used as |
| # a direct dependency. |
| # TODO(b/199420795) investigate minimal set of dependencies ... |
| py_strict_library( |
| name = "keras_lib", |
| srcs_version = "PY3", |
| visibility = [ |
| "//tensorflow:__pkg__", |
| "//tensorflow:internal", |
| "//tensorflow/python/estimator:__subpackages__", |
| "//tensorflow/python/keras:__subpackages__", |
| "//tensorflow/python/tools:__pkg__", |
| "//tensorflow/python/tools/api/generator:__pkg__", |
| "//tensorflow/tools/api/tests:__pkg__", |
| "//tensorflow/tools/compatibility/update:__pkg__", |
| "//tensorflow_estimator:__subpackages__", |
| "//third_party/py/tensorflow_privacy:__subpackages__", # TODO(b/163395075): remove when fixed |
| ], |
| deps = [ |
| "//tensorflow/python/feature_column:feature_column_py", |
| "//tensorflow/python/keras", |
| "//tensorflow/python/layers", |
| "//tensorflow/python/ops:rnn", |
| ], |
| ) |
| |
| py_strict_library( |
| name = "no_contrib", |
| srcs = ["__init__.py"], |
| srcs_version = "PY3", |
| visibility = [ |
| "//tensorflow:__pkg__", |
| "//tensorflow/python/estimator:__subpackages__", |
| "//tensorflow/python/keras:__subpackages__", |
| "//tensorflow/python/tools:__pkg__", |
| "//tensorflow/python/tools/api/generator:__pkg__", |
| "//tensorflow/tools/api/tests:__pkg__", |
| "//tensorflow/tools/compatibility/update:__pkg__", |
| "//third_party/py/keras:__subpackages__", |
| "//third_party/py/tensorflow_core:__subpackages__", |
| ], |
| deps = [ |
| ":_pywrap_py_exception_registry", |
| ":_pywrap_quantize_training", |
| ":distributed_framework_test_lib", |
| ":keras_lib", |
| ":proto_exports", |
| ":pywrap_tensorflow", |
| ":pywrap_tfe", |
| "//tensorflow/compiler/mlir/quantization/tensorflow/python:quantize_model", |
| "//tensorflow/core:protos_all_py", |
| "//tensorflow/dtensor/python:dtensor", |
| "//tensorflow/python/autograph", |
| "//tensorflow/python/autograph/utils:testing", |
| "//tensorflow/python/client", |
| "//tensorflow/python/client:_pywrap_events_writer", |
| "//tensorflow/python/client:pywrap_tf_session", |
| "//tensorflow/python/compat", |
| "//tensorflow/python/compat:v2_compat", |
| "//tensorflow/python/compiler/mlir", |
| "//tensorflow/python/compiler/tensorrt:init_py", |
| "//tensorflow/python/compiler/xla:compiler_py", |
| "//tensorflow/python/data", |
| "//tensorflow/python/debug:debug_py", |
| "//tensorflow/python/distribute", |
| "//tensorflow/python/distribute:combinations", # For tf.__internal__ API. |
| "//tensorflow/python/distribute:distribute_config", |
| "//tensorflow/python/distribute:estimator_training", |
| "//tensorflow/python/distribute:strategy_combinations", # For tf.__internal__, |
| "//tensorflow/python/distribute/experimental/rpc:rpc_ops", |
| "//tensorflow/python/dlpack", |
| "//tensorflow/python/eager:def_function", |
| "//tensorflow/python/eager:monitoring", |
| "//tensorflow/python/eager:profiler", |
| "//tensorflow/python/eager:profiler_client", |
| "//tensorflow/python/eager:remote", |
| "//tensorflow/python/framework", |
| "//tensorflow/python/framework:_pywrap_python_op_gen", |
| "//tensorflow/python/framework:_test_metrics_util", |
| "//tensorflow/python/framework:combinations", |
| "//tensorflow/python/framework:config", |
| "//tensorflow/python/framework:errors", |
| "//tensorflow/python/framework:extension_type", |
| "//tensorflow/python/framework:flexible_dtypes", |
| "//tensorflow/python/framework:for_generated_wrappers", |
| "//tensorflow/python/framework:graph_util", |
| "//tensorflow/python/framework:kernels", |
| "//tensorflow/python/framework:subscribe", |
| "//tensorflow/python/framework:tensor_spec", |
| "//tensorflow/python/framework:test_ops", # TODO(b/183988750): Break testing code out into separate rule. |
| "//tensorflow/python/framework:weak_tensor", |
| "//tensorflow/python/grappler:tf_cluster", |
| "//tensorflow/python/grappler:tf_item", |
| "//tensorflow/python/grappler:tf_optimizer", |
| "//tensorflow/python/lib/io:lib", |
| "//tensorflow/python/module", |
| "//tensorflow/python/ops:array_ops", |
| "//tensorflow/python/ops:array_ops_stack", |
| "//tensorflow/python/ops:audio_ops_gen", |
| "//tensorflow/python/ops:bincount_ops", |
| "//tensorflow/python/ops:bitwise_ops", |
| "//tensorflow/python/ops:boosted_trees_ops", |
| "//tensorflow/python/ops:check_ops", |
| "//tensorflow/python/ops:clustering_ops", |
| "//tensorflow/python/ops:collective_ops", |
| "//tensorflow/python/ops:composite_tensor_ops", |
| "//tensorflow/python/ops:cond", |
| "//tensorflow/python/ops:cond_v2", |
| "//tensorflow/python/ops:confusion_matrix", |
| "//tensorflow/python/ops:control_flow_assert", |
| "//tensorflow/python/ops:control_flow_case", |
| "//tensorflow/python/ops:control_flow_ops", |
| "//tensorflow/python/ops:control_flow_switch_case", |
| "//tensorflow/python/ops:cudnn_rnn_ops_gen", |
| "//tensorflow/python/ops:filesystem_ops", |
| "//tensorflow/python/ops:functional_ops", |
| "//tensorflow/python/ops:gradient_checker", |
| "//tensorflow/python/ops:gradient_checker_v2", |
| "//tensorflow/python/ops:histogram_ops", |
| "//tensorflow/python/ops:image_ops", |
| "//tensorflow/python/ops:initializers_ns", |
| "//tensorflow/python/ops:io_ops", |
| "//tensorflow/python/ops:list_ops", |
| "//tensorflow/python/ops:manip_ops", |
| "//tensorflow/python/ops:map_fn", |
| "//tensorflow/python/ops:map_ops", |
| "//tensorflow/python/ops:math_ops", |
| "//tensorflow/python/ops:metrics", |
| "//tensorflow/python/ops:nccl_ops", |
| "//tensorflow/python/ops:nn", |
| "//tensorflow/python/ops:proto_ops", |
| "//tensorflow/python/ops:random_crop_ops", |
| "//tensorflow/python/ops:ref_variable", |
| "//tensorflow/python/ops:rnn_cell", |
| "//tensorflow/python/ops:rnn_ops_gen", |
| "//tensorflow/python/ops:script_ops", |
| "//tensorflow/python/ops:sendrecv_ops_gen", |
| "//tensorflow/python/ops:session_ops", |
| "//tensorflow/python/ops:sets", |
| "//tensorflow/python/ops:sets_impl", |
| "//tensorflow/python/ops:sparse_ops", |
| "//tensorflow/python/ops:standard_ops", |
| "//tensorflow/python/ops:state_ops", |
| "//tensorflow/python/ops:stateful_random_ops", |
| "//tensorflow/python/ops:string_ops", |
| "//tensorflow/python/ops:tensor_array_ops", |
| "//tensorflow/python/ops:uniform_quant_ops_gen", |
| "//tensorflow/python/ops:variable_v1", |
| "//tensorflow/python/ops:weak_tensor_ops", |
| "//tensorflow/python/ops:weak_tensor_test_util", |
| "//tensorflow/python/ops:weights_broadcast_ops", |
| "//tensorflow/python/ops:while_loop", |
| "//tensorflow/python/ops:while_v2", |
| "//tensorflow/python/ops/distributions", |
| "//tensorflow/python/ops/distributions:bijector_test_util", |
| "//tensorflow/python/ops/distributions:identity_bijector", |
| "//tensorflow/python/ops/linalg", |
| "//tensorflow/python/ops/linalg:linear_operator_test_util", |
| "//tensorflow/python/ops/linalg/sparse:sparse_py", |
| "//tensorflow/python/ops/losses", |
| "//tensorflow/python/ops/numpy_ops:np_config", |
| "//tensorflow/python/ops/numpy_ops:np_random", |
| "//tensorflow/python/ops/numpy_ops:numpy", |
| "//tensorflow/python/ops/parallel_for", |
| "//tensorflow/python/ops/ragged", |
| "//tensorflow/python/ops/signal", |
| "//tensorflow/python/ops/signal:dct_ops", |
| "//tensorflow/python/ops/signal:fft_ops", |
| "//tensorflow/python/ops/signal:mel_ops", |
| "//tensorflow/python/ops/signal:mfcc_ops", |
| "//tensorflow/python/ops/signal:reconstruction_ops", |
| "//tensorflow/python/ops/signal:shape_ops", |
| "//tensorflow/python/ops/signal:spectral_ops", |
| "//tensorflow/python/ops/signal:util_ops", |
| "//tensorflow/python/ops/signal:window_ops", |
| "//tensorflow/python/ops/structured:structured_ops", |
| "//tensorflow/python/platform:_pywrap_stacktrace_handler", |
| "//tensorflow/python/platform:app", |
| "//tensorflow/python/platform:client_testlib", |
| "//tensorflow/python/platform:flags", |
| "//tensorflow/python/platform:gfile", |
| "//tensorflow/python/platform:resource_loader", |
| "//tensorflow/python/platform:sysconfig", |
| "//tensorflow/python/platform:tf_logging", |
| "//tensorflow/python/profiler", |
| "//tensorflow/python/profiler:profiler_client", |
| "//tensorflow/python/profiler:profiler_v2", |
| "//tensorflow/python/profiler:trace", |
| "//tensorflow/python/saved_model", |
| "//tensorflow/python/summary:summary_py", |
| "//tensorflow/python/summary/writer", |
| "//tensorflow/python/tools:module_util", |
| "//tensorflow/python/tools/api/generator:create_python_api", |
| "//tensorflow/python/tpu:datasets", |
| "//tensorflow/python/tpu:functional", |
| "//tensorflow/python/tpu:preempted_hook_py", |
| "//tensorflow/python/tpu:tpu_noestimator", |
| "//tensorflow/python/training:evaluation", |
| "//tensorflow/python/training:quantize_training", |
| "//tensorflow/python/training:saver", |
| "//tensorflow/python/training:saver_test_utils", |
| "//tensorflow/python/types:core", |
| "//tensorflow/python/types:data", |
| "//tensorflow/python/types:distribute", |
| "//tensorflow/python/types:trace", |
| "//tensorflow/python/user_ops:ops", |
| "//tensorflow/python/util:_pywrap_checkpoint_reader", |
| "//tensorflow/python/util:_pywrap_kernel_registry", |
| "//tensorflow/python/util:_pywrap_nest", |
| "//tensorflow/python/util:_pywrap_stat_summarizer", |
| "//tensorflow/python/util:_pywrap_tfprof", |
| "//tensorflow/python/util:_pywrap_transform_graph", |
| "//tensorflow/python/util:_pywrap_util_port", |
| "//tensorflow/python/util:_pywrap_utils", |
| "//tensorflow/python/util:all_util", |
| "//tensorflow/python/util:compat", |
| "//tensorflow/python/util:dispatch", |
| "//tensorflow/python/util:module_wrapper", |
| "//tensorflow/python/util:tf_decorator_export", |
| "//tensorflow/python/util:tf_export", |
| "//third_party/py/numpy", |
| ], |
| ) |
| |
| py_strict_library( |
| name = "core", |
| visibility = [ |
| "//tensorflow:__pkg__", |
| ], |
| deps = [ |
| "//tensorflow/python/types:core", |
| "//tensorflow/python/util:core", |
| ], |
| ) |
| |
| # This target should only be used for API generation. |
| py_strict_library( |
| name = "modules_with_exports", |
| srcs = ["modules_with_exports.py"], |
| srcs_version = "PY3", |
| visibility = [ |
| "//tensorflow:__pkg__", |
| "//tensorflow/python/tools/api/generator:__pkg__", |
| "//tensorflow/tools/compatibility/update:__pkg__", |
| "//third_party/py/tensorflow_core:__subpackages__", |
| ], |
| deps = [ |
| ":no_contrib", |
| ":proto_exports", |
| ":pywrap_tensorflow", |
| ":tf2", |
| "//tensorflow/core:protos_all_py", |
| "//tensorflow/core/function/trace_type", |
| "//tensorflow/python/client", |
| "//tensorflow/python/compat:v2_compat", |
| "//tensorflow/python/compiler/mlir", |
| "//tensorflow/python/compiler/xla", |
| "//tensorflow/python/compiler/xla:compiler_py", |
| "//tensorflow/python/data", |
| "//tensorflow/python/debug/lib:check_numerics_callback", |
| "//tensorflow/python/debug/lib:dumping_callback", |
| "//tensorflow/python/distribute", |
| "//tensorflow/python/distribute:combinations", |
| "//tensorflow/python/distribute:merge_call_interim", |
| "//tensorflow/python/distribute:multi_process_runner", |
| "//tensorflow/python/distribute:multi_worker_test_base", |
| "//tensorflow/python/distribute:parameter_server_strategy_v2", |
| "//tensorflow/python/distribute:sharded_variable", |
| "//tensorflow/python/distribute:strategy_combinations", |
| "//tensorflow/python/distribute/coordinator:cluster_coordinator", |
| "//tensorflow/python/distribute/experimental/rpc:rpc_ops", |
| "//tensorflow/python/distribute/failure_handling:failure_handling_lib", |
| "//tensorflow/python/distribute/failure_handling:preemption_watcher", |
| "//tensorflow/python/dlpack", |
| "//tensorflow/python/eager:context", |
| "//tensorflow/python/eager:def_function", |
| "//tensorflow/python/eager:monitoring", |
| "//tensorflow/python/eager:remote", |
| "//tensorflow/python/feature_column:feature_column_py", |
| "//tensorflow/python/framework", |
| "//tensorflow/python/framework:combinations", |
| "//tensorflow/python/framework:composite_tensor", |
| "//tensorflow/python/framework:config", |
| "//tensorflow/python/framework:errors", |
| "//tensorflow/python/framework:extension_type", |
| "//tensorflow/python/framework:graph_util", |
| "//tensorflow/python/framework:ops", |
| "//tensorflow/python/framework:test_combinations_lib", |
| "//tensorflow/python/framework:versions", |
| "//tensorflow/python/lib/io:lib", |
| "//tensorflow/python/module", |
| "//tensorflow/python/ops:audio_ops_gen", |
| "//tensorflow/python/ops:bincount_ops", |
| "//tensorflow/python/ops:bitwise_ops", |
| "//tensorflow/python/ops:boosted_trees_ops_gen", |
| "//tensorflow/python/ops:clustering_ops_gen", |
| "//tensorflow/python/ops:composite_tensor_ops", |
| "//tensorflow/python/ops:cond_v2", |
| "//tensorflow/python/ops:cudnn_rnn_ops_gen", |
| "//tensorflow/python/ops:debug_ops_gen", |
| "//tensorflow/python/ops:filesystem_ops_gen", |
| "//tensorflow/python/ops:gradient_checker_v2", |
| "//tensorflow/python/ops:image_ops", |
| "//tensorflow/python/ops:initializers_ns", |
| "//tensorflow/python/ops:manip_ops", |
| "//tensorflow/python/ops:map_ops_gen", |
| "//tensorflow/python/ops:metrics", |
| "//tensorflow/python/ops:nn", |
| "//tensorflow/python/ops:random_crop_ops", |
| "//tensorflow/python/ops:rnn", |
| "//tensorflow/python/ops:rnn_cell", |
| "//tensorflow/python/ops:rnn_ops_gen", |
| "//tensorflow/python/ops:sendrecv_ops_gen", |
| "//tensorflow/python/ops:sets", |
| "//tensorflow/python/ops:standard_ops", |
| "//tensorflow/python/ops:stateful_random_ops", |
| "//tensorflow/python/ops:tpu_ops_gen", |
| "//tensorflow/python/ops:uniform_quant_ops_gen", |
| "//tensorflow/python/ops:while_v2", |
| "//tensorflow/python/ops/linalg", |
| "//tensorflow/python/ops/linalg/sparse:sparse_py", |
| "//tensorflow/python/ops/losses", |
| "//tensorflow/python/ops/numpy_ops:np_array_ops", |
| "//tensorflow/python/ops/numpy_ops:np_arrays", |
| "//tensorflow/python/ops/numpy_ops:np_config", |
| "//tensorflow/python/ops/numpy_ops:np_dtypes", |
| "//tensorflow/python/ops/numpy_ops:np_math_ops", |
| "//tensorflow/python/ops/numpy_ops:np_random", |
| "//tensorflow/python/ops/numpy_ops:np_utils", |
| "//tensorflow/python/ops/ragged", |
| "//tensorflow/python/ops/ragged:ragged_ops", |
| "//tensorflow/python/ops/signal", |
| "//tensorflow/python/ops/structured:structured_ops", |
| "//tensorflow/python/platform:app", |
| "//tensorflow/python/platform:client_testlib", |
| "//tensorflow/python/platform:flags", |
| "//tensorflow/python/platform:gfile", |
| "//tensorflow/python/platform:resource_loader", |
| "//tensorflow/python/platform:sysconfig", |
| "//tensorflow/python/platform:tf_logging", |
| "//tensorflow/python/profiler", |
| "//tensorflow/python/profiler:profiler_client", |
| "//tensorflow/python/profiler:profiler_v2", |
| "//tensorflow/python/profiler:trace", |
| "//tensorflow/python/saved_model", |
| "//tensorflow/python/summary:summary_py", |
| "//tensorflow/python/training", |
| "//tensorflow/python/training:quantize_training", |
| "//tensorflow/python/user_ops:ops", |
| "//tensorflow/python/util:all_util", |
| "//tensorflow/python/util:compat", |
| "//tensorflow/python/util:dispatch", |
| "//tensorflow/python/util:tf_decorator", |
| "//tensorflow/python/util:tf_decorator_export", |
| "//tensorflow/python/util:tf_export", |
| ], |
| ) |
| |
| # Necessary for the pywrap inclusion below. |
| tf_pybind_cc_library_wrapper( |
| name = "tfcompile_headers_lib", |
| compatible_with = [], |
| deps = [ |
| "//tensorflow/compiler/aot:tfcompile_lib", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_tfcompile", |
| srcs = ["tfcompile_wrapper.cc"], |
| dynamic_deps = [":_pywrap_tensorflow_internal.so"] + select({ |
| "//tensorflow:macos": ["//tensorflow:libtensorflow_framework.%s.dylib" % VERSION], |
| "//conditions:default": ["//tensorflow:libtensorflow_framework.so.%s" % VERSION], |
| "//tensorflow:windows": [], |
| }), |
| enable_stub_generation = True, |
| features = ["-layering_check"], |
| pytype_srcs = [ |
| "_pywrap_tfcompile.pyi", |
| ], |
| static_deps = tf_python_pybind_static_deps(), |
| deps = [ |
| ":tfcompile_headers_lib", |
| "@pybind11", |
| "//third_party/python_runtime:headers", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//tensorflow/python/lib/core:pybind11_status", |
| # The headers here cannot be brought in via cc_header_only_library |
| "//tensorflow/compiler/aot:llvm_targets", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_quantize_training", |
| srcs = [ |
| "//tensorflow/python/training:quantize_training_wrapper.cc", |
| ], |
| hdrs = ["//tensorflow/core/common_runtime:quantize_training_hdrs"], |
| enable_stub_generation = True, |
| pytype_srcs = [ |
| "_pywrap_quantize_training.pyi", |
| ], |
| deps = [ |
| "//tensorflow/core:framework_headers_lib", |
| "//tensorflow/core:lib_headers_for_pybind", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/common_runtime:core_cpu_headers_lib", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//tensorflow/python/lib/core:pybind11_proto", |
| "//tensorflow/python/lib/core:pybind11_status", |
| "//third_party/python_runtime:headers", |
| "@com_google_absl//absl/strings", |
| "@pybind11", |
| ], |
| ) |
| |
| # TODO(yanhuasun): Move this back and the source file back to lib/core directory. |
| tf_python_pybind_extension( |
| name = "_pywrap_py_exception_registry", |
| srcs = ["py_exception_registry_wrapper.cc"], |
| dynamic_deps = [":_pywrap_tensorflow_internal.so"] + select({ |
| "//tensorflow:macos": ["//tensorflow:libtensorflow_framework.%s.dylib" % VERSION], |
| "//conditions:default": ["//tensorflow:libtensorflow_framework.so.%s" % VERSION], |
| "//tensorflow:windows": [], |
| }), |
| static_deps = tf_python_pybind_static_deps(), |
| # Do not sort: core:py_exception_registry must come before platform:status |
| deps = [ |
| "@com_google_absl//absl/container:fixed_array", |
| "@pybind11", |
| "//third_party/python_runtime:headers", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/python/lib/core:py_exception_registry", |
| "//tensorflow/core/platform:status", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_toco_api", |
| srcs = [ |
| "lite/toco_python_api_wrapper.cc", |
| ], |
| hdrs = ["//tensorflow/lite/toco/python:toco_python_api_hdrs"], |
| enable_stub_generation = True, |
| pytype_srcs = [ |
| "_pywrap_toco_api.pyi", |
| ], |
| deps = [ |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//third_party/python_runtime:headers", |
| "@pybind11", |
| ], |
| ) |
| |
| # TODO(edloper): Remove unused dependency on safe_ptr. (Blocker: there are |
| # targets that depend are relying on cpp_python_util to pull in safe_ptr's |
| # third_party/tensorflow/c:c_api_no_xla dependency, which registers |
| # ops/gradients, rather than depending on it themselves.) |
| cc_header_only_library( |
| name = "py_func_headers_lib", |
| features = ["-parse_headers"], |
| tags = ["no-ide"], |
| deps = [ |
| "//tensorflow/python/lib/core:py_func_lib", |
| ], |
| ) |
| |
| cc_header_only_library( |
| name = "python_op_gen_headers_lib", |
| extra_deps = [ |
| "//tensorflow/core:protos_all_cc", |
| ], |
| features = ["-parse_headers"], |
| tags = ["no-ide"], |
| deps = [ |
| "//tensorflow/python/framework:python_op_gen", |
| ], |
| ) |
| |
| py_strict_library( |
| name = "extra_py_tests_deps", |
| srcs_version = "PY3", |
| # extra_py_tests_deps is public to allow the use of tf_py_test or |
| # cuda_py_test build rules for out-of-tree custom ops. |
| # extra_py_tests_deps should NOT be used as dependency directly |
| # in out-of-tree BUILD files. |
| # Some of the dependencies in :keras_lib are possibly outdated |
| # or not generally useful, so a future task is to determine a good |
| # set of dependencies. |
| # TODO(b/199420795) investigate minimal set of dependencies ... |
| visibility = ["//visibility:public"], |
| deps = [ |
| ":keras_lib", |
| "//third_party/py/numpy", |
| "@pypi_scipy//:pkg", |
| "@six_archive//:six", |
| ], |
| ) |
| |
| py_strict_library( |
| name = "distributed_framework_test_lib", |
| srcs_version = "PY3", |
| deps = ["//tensorflow/python/framework:test_lib"], |
| ) |
| |
| # Note: this is a heavyweight library specialized for TensorFlow graphs. Do not use for |
| # other purposes. |
| |
| py_strict_library( |
| name = "global_test_configuration", |
| compatible_with = get_compatible_with_portable(), |
| srcs_version = "PY3", |
| deps = if_mlir(["//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_pass_registration"]) + |
| tf_enable_mlir_bridge(), |
| ) |
| |
| # `tree.compat` requires visibility exception to test against `nest_test` |
| # to facilitate convergence between `tree.compat` and `nest`. |
| |
| tf_proto_library( |
| name = "protos_all", |
| srcs = glob( |
| ["**/*.proto"], |
| exclude = [ |
| "//tensorflow/python/util:compare_test_proto_src", |
| "framework/cpp_shape_inference.proto", |
| ], |
| ), |
| protodeps = ["//tensorflow/python/training:checkpoint_state"], |
| visibility = visibility, |
| ) |
| |
| py_strict_library( |
| name = "pywrap_tensorflow", |
| srcs = [ |
| "pywrap_tensorflow.py", |
| ] + if_static( |
| ["pywrap_dlopen_global_flags.py"], |
| # Import will fail, indicating no global dlopen flags |
| otherwise = [], |
| ), # b/153585257 |
| srcs_version = "PY3", |
| deps = [ |
| ":pywrap_tensorflow_internal", |
| "//tensorflow/python/platform:self_check", |
| ], |
| ) |
| |
| pywrap_tensorflow_macro( |
| name = "pywrap_tensorflow_internal", |
| srcs = ["pywrap_tensorflow_internal.cc"], |
| dynamic_deps = ["//tensorflow/tsl/python/lib/core:ml_dtypes.so"] + select({ |
| "//tensorflow:macos": [ |
| "//tensorflow:libtensorflow_cc.%s.dylib" % VERSION, |
| "//tensorflow:libtensorflow_framework.%s.dylib" % VERSION, |
| ], |
| "//conditions:default": [ |
| "//tensorflow:libtensorflow_cc.so.%s" % VERSION, |
| "//tensorflow:libtensorflow_framework.so.%s" % VERSION, |
| ], |
| "//tensorflow:windows": [], |
| }), |
| exports_filter = [ |
| "@pybind11", |
| "@local_config_python//:python_headers", |
| ] + if_windows([ |
| "//:__subpackages__", |
| "@nsync//:__subpackages__", |
| "@com_google_absl//:__subpackages__", |
| "@com_google_protobuf//:__subpackages__", |
| "@double_conversion//:__subpackages__", |
| "@eigen_archive//:__subpackages__", |
| ]), |
| roots = [ |
| "//tensorflow/python/lib/core:py_exception_registry", |
| ], |
| static_deps = [ |
| "@arm_neon_2_x86_sse//:__subpackages__", |
| "@bazel_tools//:__subpackages__", |
| "@boringssl//:__subpackages__", |
| "@clog//:__subpackages__", |
| "@com_github_cares_cares//:__subpackages__", |
| "@com_github_googlecloudplatform_tensorflow_gcp_tools//:__subpackages__", |
| "@com_github_grpc_grpc//:__subpackages__", |
| "@com_google_absl//:__subpackages__", |
| "@com_google_googleapis//:__subpackages__", |
| "@com_google_protobuf//:__subpackages__", |
| "@com_googlesource_code_re2//:__subpackages__", |
| "@compute_library//:__subpackages__", |
| "@cpuinfo//:__subpackages__", |
| "@cudnn_frontend_archive//:__subpackages__", |
| "@curl//:__subpackages__", |
| "@dlpack//:__subpackages__", |
| "@double_conversion//:__subpackages__", |
| "@eigen_archive//:__subpackages__", |
| "@farmhash_archive//:__subpackages__", |
| "@farmhash_gpu_archive//:__subpackages__", |
| "@fft2d//:__subpackages__", |
| "@flatbuffers//:__subpackages__", |
| "@FP16//:__subpackages__", |
| "@FXdiv//:__subpackages__", |
| "@gemmlowp//:__subpackages__", |
| "@gif//:__subpackages__", |
| "@highwayhash//:__subpackages__", |
| "@hwloc//:__subpackages__", |
| "@icu//:__subpackages__", |
| "@jsoncpp_git//:__subpackages__", |
| "@libjpeg_turbo//:__subpackages__", |
| "@llvm_openmp//:__subpackages__", |
| "@llvm-project//:__subpackages__", |
| "@llvm_terminfo//:__subpackages__", |
| "@llvm_zlib//:__subpackages__", |
| "@local_config_cuda//:__subpackages__", |
| "@local_config_git//:__subpackages__", |
| "@local_config_nccl//:__subpackages__", |
| "@local_config_python//:__subpackages__", |
| "@local_config_rocm//:__subpackages__", |
| "@local_config_tensorrt//:__subpackages__", |
| "@local_execution_config_platform//:__subpackages__", |
| "@mkl_dnn_acl_compatible//:__subpackages__", |
| "@nccl_archive//:__subpackages__", |
| "@nsync//:__subpackages__", |
| "@nvtx_archive//:__subpackages__", |
| "@onednn//:__subpackages__", |
| "@org_sqlite//:__subpackages__", |
| "@platforms//:__subpackages__", |
| "@png//:__subpackages__", |
| "@pthreadpool//:__subpackages__", |
| "@pybind11//:__subpackages__", |
| "@ruy//:__subpackages__", |
| "@snappy//:__subpackages__", |
| "@sobol_data//:__subpackages__", |
| "@stablehlo//:__subpackages__", |
| "@tf_runtime//:__subpackages__", |
| "//:__subpackages__", |
| "@upb//:__subpackages__", |
| "@XNNPACK//:__subpackages__", |
| "@zlib//:__subpackages__", |
| ] + tsl_async_value_deps(), |
| win_def_file = ":pywrap_tensorflow_filtered_def_file", |
| deps = [ |
| "//tensorflow/c:c_api", |
| "//tensorflow/c:c_api_experimental", |
| "//tensorflow/c:checkpoint_reader", |
| "//tensorflow/c:env", |
| "//tensorflow/c:kernels", |
| "//tensorflow/c:kernels_experimental", |
| "//tensorflow/c:logging", |
| "//tensorflow/c:ops", |
| "//tensorflow/c:python_api", |
| "//tensorflow/c:safe_ptr", |
| "//tensorflow/c:tf_status_helper", |
| "//tensorflow/c/eager:c_api", |
| "//tensorflow/c/eager:c_api_experimental", |
| "//tensorflow/c/experimental/filesystem:filesystem_interface", |
| "//tensorflow/c/experimental/gradients", |
| "//tensorflow/c/experimental/gradients/tape", |
| "//tensorflow/c/experimental/ops", |
| "//tensorflow/c/experimental/stream_executor", |
| "//tensorflow/cc/saved_model:fingerprinting_impl", |
| "//tensorflow/cc/saved_model:loader_lite_impl", |
| "//tensorflow/cc/saved_model:metrics_impl", |
| "//tensorflow/compiler/mlir/python:mlir", |
| "//tensorflow/compiler/mlir/quantization/tensorflow/python:quantize_model_cc_impl", |
| "//tensorflow/compiler/mlir/tensorflow/c:mlir_c_api_registration", |
| "//tensorflow/compiler/tf2tensorrt:op_converter_registry_impl", |
| "//tensorflow/compiler/tf2xla:tf2xla_opset", |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_impl", |
| "//tensorflow/core:framework_internal_impl", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:lib_internal_impl", |
| "//tensorflow/core:reader_base", |
| "//tensorflow/core/common_runtime:core_cpu_impl", |
| "//tensorflow/core/common_runtime/gpu:gpu_runtime_impl", |
| "//tensorflow/core/common_runtime/pluggable_device:pluggable_device_runtime_impl", |
| "//tensorflow/core/config:flag_defs", |
| "//tensorflow/core/config:flags", |
| "//tensorflow/core/data/service:dispatcher_client", |
| "//tensorflow/core/data/service:grpc_util", |
| "//tensorflow/core/data/service:py_utils", |
| "//tensorflow/core/data/service:server_lib", |
| "//tensorflow/core/debug", |
| "//tensorflow/core/distributed_runtime:server_lib", |
| "//tensorflow/core/function/runtime_client:runtime_client_cc", |
| "//tensorflow/core/grappler:grappler_item", |
| "//tensorflow/core/grappler:grappler_item_builder", |
| "//tensorflow/core/grappler/clusters:cluster", |
| "//tensorflow/core/grappler/clusters:single_machine", |
| "//tensorflow/core/grappler/clusters:virtual_cluster", |
| "//tensorflow/core/grappler/costs:graph_memory", |
| "//tensorflow/core/grappler/graph_analyzer:graph_analyzer_tool", |
| "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry_impl", |
| "//tensorflow/core/grappler/optimizers:meta_optimizer", |
| "//tensorflow/core/kernels:data_service_ops", |
| "//tensorflow/core/platform:stacktrace_handler", |
| "//tensorflow/core/profiler:profiler_impl", |
| "//tensorflow/core/profiler/backends/cpu:python_tracer", |
| "//tensorflow/core/profiler/internal:print_model_analysis", |
| "//tensorflow/core/util:determinism", |
| "//tensorflow/distribute/experimental/rpc/kernels:rpc_ops", |
| "//tensorflow/dtensor/cc:dtensor_device_cc", |
| "//tensorflow/dtensor/cc:tensor_layout", |
| "//tensorflow/lite/delegates/flex:delegate", |
| "//tensorflow/lite/kernels/shim:tf_kernel_shim", |
| "//tensorflow/lite/toco/python:toco_python_api", |
| "//tensorflow/python/client:tf_session_helper", |
| "//tensorflow/python/eager:pywrap_tfe_lib", |
| "//tensorflow/python/framework:op_def_util_cc", |
| "//tensorflow/python/framework:python_api_dispatcher", |
| "//tensorflow/python/framework:python_api_info", |
| "//tensorflow/python/framework:python_api_parameter_converter", |
| "//tensorflow/python/framework:python_op_gen", |
| "//tensorflow/python/framework:python_tensor_converter", |
| "//tensorflow/python/grappler:cost_analyzer_lib", |
| "//tensorflow/python/grappler:model_analyzer_lib", |
| "//tensorflow/python/lib/core:py_func_lib", |
| "//tensorflow/python/lib/core:pybind11_absl", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//tensorflow/python/lib/core:pybind11_proto", |
| "//tensorflow/python/lib/core:pybind11_status", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr", |
| "//tensorflow/python/util:cpp_nest", |
| "//tensorflow/python/util:cpp_python_util", |
| "//tensorflow/python/util:function_parameter_canonicalizer", |
| "//tensorflow/python/util:kernel_registry", |
| "//tensorflow/tools/graph_transforms:transform_graph_lib", |
| "//tensorflow/tsl/profiler/rpc:profiler_server_impl", |
| "//tensorflow/tsl/profiler/rpc/client:profiler_client_impl", |
| "//tensorflow/tsl/python/lib/core", |
| "//third_party/python_runtime:headers", |
| "@com_google_absl//absl/base", |
| "@com_google_absl//absl/container:inlined_vector", |
| "@com_google_absl//absl/types:optional", |
| "@com_google_absl//absl/types:span", |
| ] + if_static([ |
| "//tensorflow/core/platform:tensor_float_32_utils", |
| "//tensorflow/core/platform:enable_tf2_utils", |
| ]) + if_google([ |
| "//base", |
| "//tensorflow/core/util/tensor_bundle", |
| ]) + if_oss(if_static( |
| [clean_dep("@nsync//:nsync_headers")], |
| )) + if_oss([ |
| "//tensorflow/core/distributed_runtime/rpc:grpc_server_lib", |
| "//tensorflow/core/distributed_runtime/rpc:grpc_session", |
| ]) + if_oss(if_cuda_is_configured([ |
| # TODO(tmorris): These dependencies are added to get the RPATHs for |
| # nvidia standalone wheels into pywrap_tensorflow_internal. We might be |
| # able to remove this in the future, as these stubs should already |
| # be brought in via other dependencies. |
| "//tensorflow/tsl/cuda:cudnn_stub", |
| "//tensorflow/tsl/cuda:cufft_stub", |
| "//tensorflow/tsl/cuda:nccl_rpath_stub", |
| ])) + if_xla_available([ |
| "//tensorflow/compiler/aot:tfcompile_lib", |
| ]) + tf_monitoring_python_deps() + tf_additional_plugin_deps() + tf_additional_profiler_deps() + tf_additional_binary_deps(), |
| ) |
| |
| # ** Targets for Windows build (start) ** |
| # We need the following targets to expose symbols from _pywrap_tensorflow.dll |
| |
| filegroup( |
| name = "win_lib_files_for_exported_symbols", |
| srcs = [ |
| "//tensorflow/c:checkpoint_reader", # checkpoint_reader |
| "//tensorflow/c:python_api", # tf_session |
| "//tensorflow/c:safe_ptr", # checkpoint_reader |
| "//tensorflow/c:tf_status_helper", # tfe |
| "//tensorflow/cc/saved_model:fingerprinting_impl", # SavedModel fingerprinting |
| "//tensorflow/cc/saved_model:metrics_impl", # SavedModel metrics |
| "//tensorflow/compiler/jit:flags", # tfe |
| "//tensorflow/compiler/jit:get_compiler_ir", # tfe |
| "//tensorflow/compiler/mlir/python:mlir", # mlir |
| "//tensorflow/compiler/mlir/quantization/tensorflow/python:quantize_model_cc_impl", # quantization |
| "//tensorflow/compiler/tf2xla:tf2xla_opset", # pywrap_xla_ops |
| "//tensorflow/compiler/xla/stream_executor:stream_executor_pimpl", # stat_summarizer |
| "//tensorflow/core:framework_internal_impl", # op_def_registry |
| "//tensorflow/core:lib_internal_impl", # device_lib |
| "//tensorflow/core:op_gen_lib", # tf_session |
| "//tensorflow/core/common_runtime:graph_constructor", # tf_session |
| "//tensorflow/core/common_runtime:quantize_training", # quantize_training |
| "//tensorflow/core/common_runtime:session_options", # device_lib, tfe, tf_session |
| "//tensorflow/core/common_runtime:session_state", # tf_session |
| "//tensorflow/core/common_runtime/eager:context", # tfe |
| "//tensorflow/core/common_runtime/eager:eager_executor", # tfe |
| "//tensorflow/core/common_runtime/eager:tensor_handle", # tfe |
| "//tensorflow/core/config:flag_defs", # flags_api |
| "//tensorflow/core/config:flags", # flags_api |
| "//tensorflow/core/data/service:dispatcher_client", # dispatcher_client |
| "//tensorflow/core/data/service:grpc_util", # grpc_util |
| "//tensorflow/core/data/service:server_lib", # server_lib |
| "//tensorflow/core/framework:attr_value_proto_cc", # tf_text |
| "//tensorflow/core/framework:op_def_proto_cc", # tf_text |
| "//tensorflow/core/function/runtime_client:runtime_client_cc", # runtime_client_pybind |
| "//tensorflow/core/grappler:devices", # tf_cluster |
| "//tensorflow/core/grappler:grappler_item", # tf_item |
| "//tensorflow/core/grappler:grappler_item_builder", # tf_item |
| "//tensorflow/core/grappler/clusters:cluster", # tf_cluster |
| "//tensorflow/core/grappler/clusters:single_machine", # tf_cluster |
| "//tensorflow/core/grappler/clusters:utils", # tf_optimizer |
| "//tensorflow/core/grappler/clusters:virtual_cluster", # tf_cluster |
| "//tensorflow/core/grappler/costs:analytical_cost_estimator", # cost analyzer |
| "//tensorflow/core/grappler/costs:graph_memory", # tf_cluster |
| "//tensorflow/core/grappler/costs:graph_properties", # tf_item |
| "//tensorflow/core/grappler/costs:measuring_cost_estimator", # tf_cluster |
| "//tensorflow/core/grappler/costs:op_level_cost_estimator", # tf_cluster |
| "//tensorflow/core/grappler/costs:utils", # tf_cluster |
| "//tensorflow/core/grappler/graph_analyzer:graph_analyzer_tool", # graph_analyzer |
| "//tensorflow/core/grappler/optimizers:meta_optimizer", # tf_optimizer |
| "//tensorflow/core/grappler/utils:topological_sort", # tf_item |
| "//tensorflow/core/platform:cpu_feature_guard", # cpu_feature_guard |
| "//tensorflow/core/platform:statusor", # tfe |
| "//tensorflow/core/profiler/backends/cpu:traceme_recorder_impl", # profiler |
| "//tensorflow/core/profiler/internal:print_model_analysis", # tfprof |
| "//tensorflow/core/profiler/rpc/client:profiler_client_impl", # profiler |
| "//tensorflow/core/util:determinism", # determinism |
| "//tensorflow/core/util:port", # util_port |
| "//tensorflow/core/util/tensor_bundle", # checkpoint_reader |
| "//tensorflow/dtensor/cc:dtensor_device_cc", # DTensor |
| "//tensorflow/dtensor/cc:tensor_layout", # DTensor |
| "//tensorflow/lite/kernels/shim:shape", # tf_text |
| "//tensorflow/lite/kernels/shim:tf_op_shim", # tf_text |
| "//tensorflow/lite/toco/python:toco_python_api", # toco |
| "//tensorflow/python/client:tf_session_helper", # tf_session |
| "//tensorflow/python/eager:pywrap_tfe_lib", # pywrap_tfe_lib |
| "//tensorflow/python/framework:op_def_util_cc", # op_def_util |
| "//tensorflow/python/framework:python_api_dispatcher", # python_api_dispatcher |
| "//tensorflow/python/framework:python_api_info", # python_api_info |
| "//tensorflow/python/framework:python_api_parameter_converter", # python_api_parameter_converter |
| "//tensorflow/python/framework:python_op_gen", # python_op_gen |
| "//tensorflow/python/framework:python_tensor_converter", # python_tensor_converter |
| "//tensorflow/python/grappler:cost_analyzer_lib", |
| "//tensorflow/python/grappler:model_analyzer_lib", # model_analyzer |
| "//tensorflow/python/lib/core:ndarray_tensor", # checkpoint_reader |
| "//tensorflow/python/lib/core:py_exception_registry", # py_exception_registry |
| "//tensorflow/python/lib/core:py_func_lib", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr", # checkpoint_reader |
| "//tensorflow/python/util:cpp_nest", |
| "//tensorflow/python/util:cpp_python_util", |
| "//tensorflow/python/util:kernel_registry", |
| "//tensorflow/tools/graph_transforms:transform_graph_lib", # transform_graph |
| "//tensorflow/tsl/platform:tensor_float_32_utils", # tensor_float_32 |
| "//tensorflow/tsl/profiler/lib:profiler_session_impl", # profiler |
| "//tensorflow/tsl/profiler/rpc:profiler_server_impl", # profiler |
| "//tensorflow/tsl/profiler/rpc/client:profiler_client_impl", |
| "//tensorflow/tsl/python/lib/core:ml_dtypes_lib", # bfloat16, float8_e4m3fn, float8_e5m2 |
| "//tensorflow/tsl/python/lib/core:numpy", # checkpoint_reader |
| ] + if_xla_available([ |
| "//tensorflow/compiler/aot:tfcompile_lib", # tfcompile |
| "//tensorflow/compiler/xla:status_macros", # tfcompile |
| "//tensorflow/compiler/xla/hlo/ir:hlo", # tfcompile |
| ]), |
| visibility = ["//visibility:private"], |
| ) |
| |
| # Filter the DEF file to reduce the number of symbols to 64K or less. |
| # Note that we also write the name of the pyd file into DEF file so that |
| # the dynamic libraries of custom ops can find it at runtime. |
| genrule( |
| name = "pywrap_tensorflow_filtered_def_file", |
| srcs = select({ |
| "//tensorflow:windows": [ |
| ":pybind_symbol_target_libs_file", |
| ":win_lib_files_for_exported_symbols", |
| "//tensorflow:tensorflow_def_file", |
| "//tensorflow/tools/def_file_filter:symbols_pybind", |
| ], |
| "//conditions:default": [], |
| }), |
| outs = ["pywrap_tensorflow_filtered_def_file.def"], |
| cmd = select({ |
| "//tensorflow:windows": """ |
| $(location @local_config_def_file_filter//:def_file_filter) \\ |
| --input $(location //tensorflow:tensorflow_def_file) \\ |
| --output $@ \\ |
| --target _pywrap_tensorflow_internal.pyd \\ |
| --symbols $(location //tensorflow/tools/def_file_filter:symbols_pybind) \\ |
| --lib_paths_file $(location :pybind_symbol_target_libs_file) |
| """, |
| "//conditions:default": "touch $@", # Just a placeholder for Unix platforms |
| }), |
| tools = ["@local_config_def_file_filter//:def_file_filter"], |
| visibility = ["//visibility:public"], |
| ) |
| |
| # Write to a file a list of all cc_library targets that we need for exporting symbols on Windows. |
| genrule( |
| name = "pybind_symbol_target_libs_file", |
| srcs = [":win_lib_files_for_exported_symbols"], |
| outs = ["pybind_symbol_target_libs_file.txt"], |
| cmd = select({ |
| "//tensorflow:windows": """ |
| for SRC in $(SRCS); do |
| echo $$SRC | sed 's/third_party\\///g' >> $@ |
| done |
| """, |
| "//conditions:default": "touch $@", # Just a placeholder for Unix platforms |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| # Get the import library of _pywrap_tensorflow_internal.pyd, platform-specific to Windows. |
| filegroup( |
| name = "get_pywrap_tensorflow_import_lib_file", |
| srcs = [":_pywrap_tensorflow_internal.so"], |
| output_group = "interface_library", |
| ) |
| |
| cc_import( |
| name = "_pywrap_tensorflow_internal_linux", |
| shared_library = "//tensorflow/python:lib_pywrap_tensorflow_internal.so", |
| visibility = tf_external_workspace_visible(visibility), |
| ) |
| |
| cc_import( |
| name = "_pywrap_tensorflow_internal_macos", |
| shared_library = "//tensorflow/python:lib_pywrap_tensorflow_internal.dylib", |
| visibility = tf_external_workspace_visible(visibility), |
| ) |
| |
| cc_import( |
| name = "_pywrap_tensorflow_internal_windows", |
| interface_library = "//tensorflow/python:pywrap_tensorflow_import_lib_file", |
| shared_library = "//tensorflow/python:_pywrap_tensorflow_internal.dll", |
| visibility = tf_external_workspace_visible(visibility), |
| ) |
| |
| # Rename the import library for _pywrap_tensorflow_internal.pyd to _pywrap_tensorflow_internal.lib |
| # (It was _pywrap_tensorflow_internal.so.if.lib). |
| genrule( |
| name = "pywrap_tensorflow_import_lib_file", |
| srcs = [":get_pywrap_tensorflow_import_lib_file"], |
| outs = ["_pywrap_tensorflow_internal.lib"], |
| cmd = select({ |
| "//tensorflow:windows": "cp -f $< $@", |
| "//conditions:default": "touch $@", # Just a placeholder for Unix platforms |
| }), |
| visibility = ["//visibility:public"], |
| ) |
| |
| # Create a cc_import rule for the import library of _pywrap_tensorflow_internal.dll |
| # so that custom ops' dynamic libraries can link against it. |
| cc_import( |
| name = "pywrap_tensorflow_import_lib", |
| interface_library = select({ |
| "//tensorflow:windows": ":pywrap_tensorflow_import_lib_file", |
| "//conditions:default": "not_existing_on_unix.lib", # Just a placeholder for Unix platforms |
| }), |
| system_provided = 1, |
| ) |
| |
| # ** Targets for Windows build (end) ** |
| |
| py_strict_library( |
| name = "pywrap_mlir", |
| srcs = ["pywrap_mlir.py"], |
| srcs_version = "PY3", |
| visibility = ["//visibility:public"], |
| deps = [ |
| ":_pywrap_mlir", |
| ":pywrap_tensorflow", |
| "//tensorflow/python/eager:context", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_mlir", |
| srcs = ["mlir_wrapper.cc"], |
| hdrs = [ |
| "//tensorflow/c:headers", |
| "//tensorflow/c:safe_ptr_hdr", |
| "//tensorflow/c/eager:headers", |
| "//tensorflow/compiler/mlir/python:pywrap_mlir_hdrs", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr_required_hdrs", |
| ], |
| deps = [ |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/platform:status", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//tensorflow/python/lib/core:pybind11_status", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr", |
| "//third_party/python_runtime:headers", |
| "@com_google_absl//absl/strings", |
| "@pybind11", |
| ], |
| ) |
| |
| py_strict_library( |
| name = "pywrap_sanitizers", |
| srcs = ["pywrap_sanitizers.py"], |
| srcs_version = "PY3", |
| visibility = ["//visibility:public"], |
| deps = [ |
| ":_pywrap_sanitizers", |
| ":pywrap_tensorflow", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "flags_pybind", |
| srcs = ["//tensorflow/core/config:flags_api_wrapper.cc"], |
| enable_stub_generation = True, |
| pytype_srcs = [ |
| "flags_pybind.pyi", |
| ], |
| visibility = ["//tensorflow/core/config:__subpackages__"], |
| deps = [ |
| "//tensorflow/core:lib_headers_for_pybind", |
| "//tensorflow/core/config:flags_headers", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//third_party/python_runtime:headers", |
| "@com_google_absl//absl/types:optional", |
| "@pybind11", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_sanitizers", |
| srcs = ["sanitizers_wrapper.cc"], |
| enable_stub_generation = True, |
| pytype_srcs = [ |
| "_pywrap_sanitizers.pyi", |
| ], |
| deps = [ |
| "@pybind11", |
| ], |
| ) |
| |
| cc_library( |
| name = "unified_api_pywrap_required_headers", |
| textual_hdrs = [ |
| "//tensorflow/python/lib/core:basic_hdrs", |
| "//tensorflow/tsl/python/lib/core:basic_hdrs", |
| "//tensorflow/c:headers", |
| "//tensorflow/c:safe_ptr_hdr", |
| "//tensorflow/c/eager:headers", |
| "//tensorflow/c/eager:pywrap_required_hdrs", |
| "//tensorflow/c/experimental/ops:pywrap_required_hdrs", |
| "//tensorflow/c/experimental/gradients:pywrap_required_hdrs", |
| "//tensorflow/c/experimental/gradients/tape:pywrap_required_hdrs", |
| "//tensorflow/core/common_runtime/eager:pywrap_required_hdrs", |
| "//tensorflow/python/eager:pywrap_required_hdrs", |
| "//tensorflow/core/common_runtime:core_cpu_lib_headers", |
| "//tensorflow/core/public:session.h", |
| "//tensorflow/core/public:session_options.h", |
| ], |
| visibility = ["//tensorflow/python/framework/experimental:__pkg__"], |
| deps = [ |
| "//tensorflow/c:pywrap_required_hdrs", |
| "//tensorflow/core/config:flags_headers", |
| "//tensorflow/core/framework:pywrap_required_hdrs", |
| "//third_party/py/numpy:headers", |
| "//third_party/python_runtime:headers", |
| ], |
| ) |
| |
| py_strict_library( |
| name = "pywrap_tfe", |
| srcs = ["pywrap_tfe.py"], |
| srcs_version = "PY3", |
| visibility = ["//visibility:public"], |
| deps = [ |
| ":_pywrap_tfe", |
| ":pywrap_tensorflow", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_tfe", |
| srcs = ["tfe_wrapper.cc"], |
| hdrs = [ |
| "//tensorflow/c:headers", |
| "//tensorflow/c:safe_ptr_hdr", |
| "//tensorflow/c/eager:headers", |
| "//tensorflow/c/eager:pywrap_required_hdrs", |
| "//tensorflow/c/experimental/ops:pywrap_required_hdrs", |
| "//tensorflow/core/common_runtime/eager:pywrap_required_hdrs", |
| "//tensorflow/core/distributed_runtime:pywrap_required_hdrs", |
| "//tensorflow/core/distributed_runtime/coordination:pywrap_required_hdrs", |
| "//tensorflow/core/distributed_runtime/eager:pywrap_required_hdrs", |
| "//tensorflow/python/eager:pywrap_required_hdrs", |
| "//tensorflow/python/lib/core:py_exception_registry_hdr", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr_required_hdrs", |
| "//tensorflow/python/util:util_hdr", |
| "//tensorflow/tsl/distributed_runtime:pywrap_required_hdrs", |
| "//tensorflow/tsl/distributed_runtime/coordination:pywrap_required_hdrs", |
| "//tensorflow/tsl/python/lib/core:numpy_hdr", |
| ], |
| dynamic_deps = [":_pywrap_tensorflow_internal.so"] + select({ |
| "//tensorflow:macos": ["//tensorflow:libtensorflow_framework.%s.dylib" % VERSION], |
| "//conditions:default": ["//tensorflow:libtensorflow_framework.so.%s" % VERSION], |
| "//tensorflow:windows": [], |
| }), |
| static_deps = tf_python_pybind_static_deps(), |
| deps = [ |
| "//tensorflow/c:pywrap_required_hdrs", |
| "//tensorflow/c/eager:tfe_tensorhandle_internal_hdrs_only", |
| "//tensorflow/compiler/jit:flags_headers_only", |
| "//tensorflow/compiler/jit:get_compiler_ir_hdrs_only", |
| "//tensorflow/core:framework_headers_lib", |
| "//tensorflow/core:lib_headers_for_pybind", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/common_runtime:core_cpu_headers_lib", |
| "//tensorflow/core/config:flags_headers", |
| "//tensorflow/core/framework:pywrap_required_hdrs", |
| "//tensorflow/core/lib/llvm_rtti", |
| "//tensorflow/core/platform", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//tensorflow/python/lib/core:pybind11_status", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr", |
| "//third_party/py/numpy:headers", |
| "//third_party/python_runtime:headers", |
| "@com_google_absl//absl/container:flat_hash_map", |
| "@com_google_absl//absl/hash", |
| "@com_google_absl//absl/memory", |
| "@com_google_absl//absl/strings", |
| "@com_google_absl//absl/strings:str_format", |
| "@com_google_absl//absl/types:optional", |
| "@pybind11", |
| ] + if_static( |
| extra_deps = [ |
| "//tensorflow/core/protobuf:eager_service_proto_cc", |
| "//tensorflow/core/protobuf:master_proto_cc", |
| "//tensorflow/core/protobuf:worker_proto_cc", |
| "//tensorflow/tsl/protobuf:coordination_service_proto_cc", |
| ], |
| otherwise = [ |
| "//tensorflow/core/protobuf:eager_service_proto_cc_headers_only", |
| "//tensorflow/core/protobuf:master_proto_cc_headers_only", |
| "//tensorflow/core/protobuf:worker_proto_cc_headers_only", |
| "//tensorflow/tsl/protobuf:coordination_service_proto_cc_headers_only", |
| ], |
| ), |
| ) |
| |
| py_strict_library( |
| name = "pywrap_tfe_monitoring_reader", |
| testonly = True, |
| srcs = ["pywrap_tfe_monitoring_reader.py"], |
| srcs_version = "PY3", |
| visibility = ["//visibility:public"], |
| deps = [ |
| ":_pywrap_tfe_monitoring_reader", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_tfe_monitoring_reader", |
| testonly = True, |
| srcs = ["tfe_wrapper_monitoring_reader.cc"], |
| hdrs = [ |
| "//tensorflow/c/eager:headers_monitoring_reader", |
| "//tensorflow/c/eager:pywrap_headers_monitoring_reader", |
| "//tensorflow/python/util:util_hdr", |
| ], |
| enable_stub_generation = True, |
| pytype_srcs = [ |
| "_pywrap_tfe_monitoring_reader.pyi", |
| ], |
| deps = [ |
| "//tensorflow/c:c_api", |
| "//tensorflow/c/eager:c_api", |
| "//tensorflow/c/eager:c_api_experimental_reader", |
| "//tensorflow/c/eager:tfe_tensorhandle_internal_hdrs_only", |
| "//tensorflow/compiler/jit:flags_headers_only", |
| "//tensorflow/compiler/jit:get_compiler_ir_hdrs_only", |
| "//tensorflow/core:framework_headers_lib", |
| "//tensorflow/core:lib_headers_for_pybind", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/common_runtime:core_cpu_headers_lib", |
| "//tensorflow/core/config:flags_headers", |
| "//tensorflow/core/framework:pywrap_required_hdrs", |
| "//tensorflow/core/lib/llvm_rtti", |
| "//tensorflow/core/platform", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//tensorflow/python/lib/core:pybind11_status", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr", |
| "//third_party/py/numpy:headers", |
| "//third_party/python_runtime:headers", |
| "@com_google_absl//absl/container:flat_hash_map", |
| "@com_google_absl//absl/hash", |
| "@com_google_absl//absl/memory", |
| "@com_google_absl//absl/strings", |
| "@com_google_absl//absl/strings:str_format", |
| "@com_google_absl//absl/types:optional", |
| "@pybind11", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_parallel_device", |
| srcs = [ |
| "//tensorflow/c:headers", |
| "//tensorflow/c:safe_ptr_hdr", |
| "//tensorflow/c/eager:headers", |
| "//tensorflow/c/eager/parallel_device:headers", |
| "//tensorflow/c/eager/parallel_device:sources", |
| "//tensorflow/python/distribute/parallel_device:pywrap_parallel_device.cc", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr_required_hdrs", |
| ], |
| enable_stub_generation = True, |
| pytype_srcs = [ |
| "_pywrap_parallel_device.pyi", |
| ], |
| visibility = ["//tensorflow/python/distribute/parallel_device:__pkg__"], |
| deps = [ |
| "//tensorflow/c:pywrap_required_hdrs", |
| "//tensorflow/c/eager:tfe_cancellationmanager_internal_hdrs_only", |
| "//tensorflow/c/eager:tfe_op_internal", |
| "//tensorflow/c/eager:tfe_tensorhandle_internal_hdrs_only", |
| "//tensorflow/core:framework_headers_lib", |
| "//tensorflow/core:lib_headers_for_pybind", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//tensorflow/python/lib/core:pybind11_status", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr", |
| "//third_party/python_runtime:headers", |
| "@pybind11", |
| ], |
| ) |
| |
| tf_python_pybind_extension( |
| name = "_pywrap_dtensor_device", |
| srcs = ["pywrap_dtensor_device.cc"], |
| features = ["-layering_check"], |
| deps = [ |
| ":pywrap_densor_device_headers", |
| "//tensorflow/dtensor/proto:layout_proto_cc", |
| "//tensorflow/python/lib/core:pybind11_lib", |
| "//tensorflow/python/lib/core:pybind11_status_headers", |
| "//third_party/python_runtime:headers", # buildcleaner: keep |
| "@pybind11", |
| "@pybind11_abseil//pybind11_abseil:absl_casters", |
| "@pybind11_protobuf//pybind11_protobuf:native_proto_caster", |
| ], |
| ) |
| |
| tf_pybind_cc_library_wrapper( |
| name = "pywrap_densor_device_headers", |
| deps = [ |
| "//tensorflow/c:c_api", |
| "//tensorflow/c/eager:c_api", |
| "//tensorflow/c/eager:c_api_internal", |
| "//tensorflow/dtensor/cc:dtensor_device_cc", |
| "//tensorflow/dtensor/cc:tensor_layout", |
| "//tensorflow/python/eager:pywrap_tfe_lib", |
| "//tensorflow/python/lib/core:safe_pyobject_ptr", |
| ], |
| ) |
| |
| py_strict_library( |
| name = "tf2", |
| srcs = ["tf2.py"], |
| srcs_version = "PY3", |
| deps = [ |
| "//tensorflow/python/platform:_pywrap_tf2", |
| "//tensorflow/python/util:tf_export", |
| ], |
| ) |
| |
| # copybara:uncomment_begin(google-only) |
| # # TODO(b/287346201): Look into replacing usages of this target. |
| # py_proto_library( |
| # name = "protos_all_py_pb2", |
| # has_services = 0, |
| # api_version = 2, |
| # deps = [":protos_all"], |
| # ) |
| # |
| # # Special build rule for jax2tf. |
| # # TODO(b/239052279): remove once TF->TFLite dependency cycle is resolved |
| # py_strict_library( |
| # name = "jax2tf_support", |
| # visibility = ["//third_party/py/jax/experimental/jax2tf:__subpackages__"], |
| # deps = [ |
| # "//tensorflow/python/dlpack", |
| # "//tensorflow/python/ops/ragged:ragged_array_ops", |
| # "//tensorflow/python/saved_model", |
| # ], |
| # ) |
| # copybara:uncomment_end |
| |
| pytype_strict_library( |
| name = "proto_exports", |
| srcs = ["proto_exports.py"], |
| deps = [ |
| "//tensorflow/core:protos_all_py", |
| "//tensorflow/python/util:tf_export", |
| ], |
| ) |
