# Description:

# TensorFlow is a computational framework, primarily for use in machine

# learning applications.

load("@rules_cc//cc:cc_import.bzl", "cc_import")

# buildifier: disable=out-of-order-load

# copybara:comment_begin(oss-only)

load("//tensorflow:strict.default.bzl", "py_strict_library")

# copybara:comment_end

# buildifier: enable=out-of-order-load

load("@bazel_skylib//lib:selects.bzl", "selects")

load("@bazel_skylib//rules:common_settings.bzl", "bool_flag", "bool_setting")

load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")

load(

"//tensorflow:tensorflow.bzl",

"VERSION",

"VERSION_MAJOR",

"check_deps",

"if_google",

"if_oss",

"if_xla_available",

"pywrap_aware_tf_cc_shared_object",

"tf_custom_op_library_additional_deps_impl",

"tf_monitoring_python_deps",

"tf_native_cc_binary",

"tsl_async_value_deps",

)

load(

"//tensorflow:tensorflow.default.bzl",

"ADDITIONAL_API_INDEXABLE_SETTINGS",

"tf_cc_shared_library",

)

load(

"@xla//xla/tsl/mkl:build_defs.bzl",

"if_mkl_ml",

)

load(

"//tensorflow/core/platform:build_config.bzl",

"tf_additional_binary_deps",

)

load(

"//tensorflow/core/platform:build_config_root.bzl",

"if_static",

"tf_additional_plugin_deps",

"tf_additional_profiler_deps",

)

load("@bazel_skylib//:bzl_library.bzl", "bzl_library")

# Do not sort: Copybara rule

# copybara:uncomment_begin

# # buildifier: disable=out-of-order-load

# load("//devtools/build_cleaner/skylark:action_config_test.bzl", "action_config_test")

# load("//devtools/copybara/rules:copybara.bzl", "copybara_config_test")

# load("@rules_license//rules:license.bzl", "license")

# # buildifier: enable=out-of-order-load

# copybara:uncomment_end

# copybara:comment_begin(oss-only)

load("@bazel_skylib//rules:copy_file.bzl", "copy_file")

load(

"//tensorflow/python/tools/api/generator:api_gen.bzl",

"get_compat_files",

"get_nested_compat_files",

)

load(

"//tensorflow/python/tools/api/generator:api_init_files.bzl",

"TENSORFLOW_API_INIT_FILES",

)

load(

"//tensorflow/python/tools/api/generator:api_init_files_v1.bzl",

"TENSORFLOW_API_INIT_FILES_V1",

)

load("//tensorflow/python/tools/api/generator2:generate_api.bzl", "generate_apis")

# copybara:comment_end

PACKAGE_STATIC_DEPS = [

"//:__subpackages__",

"@gif//:__subpackages__",

"@highwayhash//:__subpackages__",

"@hwloc//:__subpackages__",

"@icu//:__subpackages__",

"@libjpeg_turbo//:__subpackages__",

"@llvm-project//:__subpackages__",

"@png//:__subpackages__",

"@sobol_data//:__subpackages__",

"@stablehlo//:__subpackages__",

"@tf_runtime//:__subpackages__",

"@bazel_tools//:__subpackages__",

"@boringssl//:__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__",

"@curl//:__subpackages__",

"@eigen_archive//:__subpackages__",

"@farmhash_archive//:__subpackages__",

"@farmhash_gpu_archive//:__subpackages__",

"@fft2d//:__subpackages__",

"@gemmlowp//:__subpackages__",

"@jsoncpp_git//:__subpackages__",

"@llvm_openmp//:__subpackages__",

"@llvm_terminfo//:__subpackages__",

"@llvm_zlib//:__subpackages__",

"@local_config_cuda//:__subpackages__",

"@local_config_git//:__subpackages__",

"@local_config_nccl//:__subpackages__",

"@local_config_rocm//:__subpackages__",

"@local_config_tensorrt//:__subpackages__",

"@mkl_dnn_acl_compatible//:__subpackages__",

"@ml_dtypes_py//ml_dtypes:__subpackages__",

"@nccl_archive//:__subpackages__",

"@net_zstd//:__subpackages__",

"@nvtx_archive//:__subpackages__",

"@onednn//:__subpackages__",

"@org_sqlite//:__subpackages__",

"@platforms//:__subpackages__",

"@snappy//:__subpackages__",

"@upb//:__subpackages__",

"@zlib//:__subpackages__",

"@dlpack//:__subpackages__",

"@arm_neon_2_x86_sse//:__subpackages__",

"@cpuinfo//:__subpackages__",

"@ruy//:__subpackages__",

"@XNNPACK//:__subpackages__",

"@pthreadpool//:__subpackages__",

"@FXdiv//:__subpackages__",

"@FP16//:__subpackages__",

"@clog//:__subpackages__",

"@flatbuffers//:__subpackages__",

"@nccl_archive//:__subpackages__",

"@triton//:__subpackages__",

"@tsl//:__subpackages__",

"@xla//:__subpackages__",

] + tsl_async_value_deps()

package(

# copybara:uncomment default_applicable_licenses = [":license"],

default_visibility = [":internal"],

)

# copybara:uncomment_begin

# license(

# name = "license",

# package_name = "tensorflow",

# )

#

# filegroup(

# name = "copybara_library",

# srcs = glob(

# [

# "*.bara.sky",

# "*.dic",

# "*.ftrcp",

# ],

# exclude = [

# "copy.bara.sky",

# ],

# ),

# visibility = ["//visibility:public"],

# )

#

# filegroup(

# name = "copybara_config",

# srcs = ["copy.bara.sky"],

# data = [

# ":copybara_library",

# "//learning/brain/testing/copybara:all_bara_sky",

# "//third_party/tsl:copybara_library",

# "//third_party/xla:copybara_library",

# ],

# visibility = ["//visibility:public"],

# )

#

# copybara_config_test(

# name = "copybara_config_test",

# config = "copy.bara.sky",

# tags = [

# "noasan",

# "nodfsan",

# "nogotsan",

# "nogpu",

# "nomsan",

# "nosan",

# "notsan",

# "noubsan",

# ],

# deps = [":copybara_config"],

# )

#

# action_config_test(

# name = "build_cleaner_spec_test",

# src = "build_cleaner_spec.textproto",

# )

#

# exports_files(srcs = ["METADATA"])

# copybara:uncomment_end

licenses(["notice"])

exports_files([

"LICENSE",

# The leakr files are used by //third_party/cloud_tpu and

# //third_party/tensorboard/google:copybara_config_test.

"leakr_badwords.dic",

"leakr_badfiles.dic",

"leakr_file_type_recipe.ftrcp",

])

# copybara:comment_begin(oss-only)

TENSORFLOW_API_INIT_FILES_V2 = (

TENSORFLOW_API_INIT_FILES +

get_compat_files(TENSORFLOW_API_INIT_FILES, 2) +

get_compat_files(TENSORFLOW_API_INIT_FILES_V1, 1) + get_nested_compat_files([

1,

2,

])

)

TENSORFLOW_API_INIT_FILES_V1 = (

TENSORFLOW_API_INIT_FILES_V1 +

get_compat_files(TENSORFLOW_API_INIT_FILES, 2) +

get_compat_files(TENSORFLOW_API_INIT_FILES_V1, 1) + get_nested_compat_files([

1,

2,

])

)

# copybara:comment_end

# Config setting used when building for products

# which requires restricted licenses to be avoided.

config_setting(

name = "no_lgpl_deps",

define_values = {"__TENSORFLOW_NO_LGPL_DEPS__": "1"},

visibility = ["//visibility:public"],

)

# Config setting that disables the default logger, only logging

# to registered TFLogSinks

config_setting(

name = "no_default_logger",

define_values = {"no_default_logger": "true"},

visibility = ["//visibility:public"],

)

config_setting(

name = "force_rtti",

define_values = {

"tf_force_rtti": "true",

},

visibility = ["//visibility:public"],

)

# Config setting for determining if we are building for Android.

config_setting(

name = "android",

constraint_values = if_google(

["@platforms//os:android"],

[],

),

values = if_oss(

{"crosstool_top": "//external:android/crosstool"},

{},

),

visibility = ["//visibility:public"],

)

config_setting(

name = "android_x86",

constraint_values =

[

"@platforms//cpu:x86_32",

"@platforms//os:android",

],

values = dict(

if_oss(

{"crosstool_top": "//external:android/crosstool"},

),

),

visibility = ["//visibility:public"],

)

config_setting(

name = "android_x86_64",

constraint_values =

[

"@platforms//cpu:x86_64",

"@platforms//os:android",

],

values = dict(

if_oss(

{"crosstool_top": "//external:android/crosstool"},

),

),

visibility = ["//visibility:public"],

)

config_setting(

name = "android_armeabi",

constraint_values =

[

"@platforms//cpu:armv6-m",

"@platforms//os:android",

],

values = dict(

if_oss(

{"crosstool_top": "//external:android/crosstool"},

),

),

visibility = ["//visibility:public"],

)

# copybara:uncomment_begin(google-only)

# config_setting(

# name = "chromiumos_x86_64",

# constraint_values = [

# "@platforms//cpu:x86_64",

# "@platforms//os:chromiumos",

# ],

# visibility = ["//visibility:public"],

# )

#

# config_setting(

# name = "chromiumos_arm64",

# constraint_values = [

# "@platforms//cpu:aarch64",

# "@platforms//os:chromiumos",

# ],

# visibility = ["//visibility:public"],

# )

#

# config_setting(

# name = "chromiumos_armv7",

# constraint_values = [

# "@platforms//cpu:armv7",

# "@platforms//os:chromiumos",

# ],

# visibility = ["//visibility:public"],

# )

# copybara:uncomment_end

config_setting(

name = "emscripten",

constraint_values = if_google(

["@platforms//os:emscripten"],

[],

),

values = if_oss(

{"crosstool_top": "//external:android/emscripten"},

{},

),

visibility = ["//visibility:public"],

)

config_setting(

name = "raspberry_pi_armeabi",

constraint_values =

[

"@platforms//cpu:armv6-m",

"@platforms//os:linux",

],

values = if_oss(

{"crosstool_top": "@local_config_arm_compiler//:toolchain"},

{},

),

visibility = ["//visibility:public"],

)

config_setting(

name = "android_arm",

constraint_values =

[

"@platforms//cpu:armv7",

"@platforms//os:android",

],

values = dict(

if_oss(

{"crosstool_top": "//external:android/crosstool"},

),

),

visibility = ["//visibility:public"],

)

config_setting(

name = "android_arm64",

constraint_values =

[

"@platforms//cpu:aarch64",

"@platforms//os:android",

],

values = dict(

if_oss(

{"crosstool_top": "//external:android/crosstool"},

),

),

visibility = ["//visibility:public"],

)

config_setting(

name = "android_mips64",

constraint_values =

[

"@platforms//cpu:mips64",

"@platforms//os:android",

],

values = if_oss(

{"crosstool_top": "//external:android/crosstool"},

{},

),

visibility = ["//visibility:public"],

)

# TODO(jakeharmon8): Remove in favor of TSL version

config_setting(

name = "windows",

constraint_values =

[

"@platforms//os:windows",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "no_tensorflow_py_deps",

define_values = {"no_tensorflow_py_deps": "true"},

visibility = ["//visibility:public"],

)

config_setting(

name = "macos_x86_64",

constraint_values =

[

"@platforms//cpu:x86_64",

"@platforms//os:macos",

],

values = {

"apple_platform_type": "macos",

},

visibility = ["//visibility:public"],

)

config_setting(

name = "macos_arm64",

constraint_values =

[

"@platforms//cpu:aarch64",

"@platforms//os:macos",

],

values = {

"apple_platform_type": "macos",

},

visibility = ["//visibility:public"],

)

# TODO(jakeharmon8): Remove in favor of TSL version

selects.config_setting_group(

name = "macos",

match_any = [

":macos_x86_64",

":macos_arm64",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "ios",

constraint_values = if_google(

["@platforms//os:ios"],

[],

),

values = if_oss(

{"apple_platform_type": "ios"},

{},

),

visibility = ["//visibility:public"],

)

# TODO(jakeharmon8): Remove in favor of TSL version

config_setting(

name = "fuchsia",

constraint_values =

["@platforms//os:fuchsia"],

visibility = ["//visibility:public"],

)

config_setting(

name = "fuchsia_x86_64",

constraint_values =

[

"@platforms//cpu:x86_64",

"@platforms//os:fuchsia",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "ios_x86_64",

constraint_values =

[

"@platforms//cpu:x86_64",

"@platforms//os:ios",

],

values = dict(

if_oss(

{"crosstool_top": "//tools/osx/crosstool:crosstool"},

),

),

visibility = ["//visibility:public"],

)

config_setting(

name = "chromiumos",

constraint_values = if_google(

["@platforms//os:chromiumos"],

[],

),

values = if_oss(

{"crosstool_top": "//external:android/chromiumos"},

{},

),

visibility = ["//visibility:public"],

)

config_setting(

name = "linux_aarch64",

constraint_values =

[

"@platforms//cpu:aarch64",

"@platforms//os:linux",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "linux_arm64_cross_compile",

define_values = {"linux_arm64_cross_compile": "true"},

visibility = ["//visibility:public"],

)

config_setting(

name = "linux_armhf",

constraint_values =

[

"@platforms//cpu:armv7e-mf",

"@platforms//os:linux",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "linux_x86_64",

constraint_values =

[

"@platforms//cpu:x86_64",

"@platforms//os:linux",

],

visibility = ["//visibility:public"],

)

# This condition takes precedence over :linux_x86_64

config_setting(

name = "linux_x86_64_no_sse",

constraint_values =

[

"@platforms//cpu:x86_64",

"@platforms//os:linux",

],

values = {

"copt": "-mno-sse4.2",

},

visibility = ["//visibility:public"],

)

# This condition takes precedence over :linux_x86_64

# TODO(b/290533709): Remove this with PJRT build rule cleanup.

config_setting(

name = "linux_x86_64_with_weightwatcher",

constraint_values =

[

"@platforms//cpu:x86_64",

"@platforms//os:linux",

],

define_values = {"tensorflow_weightwatcher": "true"},

visibility = ["//visibility:public"],

)

config_setting(

name = "linux_ppc64le",

constraint_values =

[

"@platforms//cpu:ppc64le",

"@platforms//os:linux",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "linux_s390x",

constraint_values =

[

"@platforms//cpu:s390x",

"@platforms//os:linux",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "linux_mips64",

constraint_values =

[

"@platforms//cpu:mips64",

"@platforms//os:linux",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "linux_riscv64",

constraint_values =

[

"@platforms//cpu:riscv64",

"@platforms//os:linux",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "debug",

values = {

"compilation_mode": "dbg",

},

visibility = ["//visibility:public"],

)

config_setting(

name = "optimized",

values = {

"compilation_mode": "opt",

},

visibility = ["//visibility:public"],

)

selects.config_setting_group(

name = "arm_any",

match_any = [

"@platforms//cpu:aarch32",

"@platforms//cpu:aarch64",

"@platforms//cpu:armv6-m",

"@platforms//cpu:armv7",

"@platforms//cpu:armv7-m",

"@platforms//cpu:armv7e-m",

"@platforms//cpu:armv7e-mf",

],

)

config_setting(

name = "freebsd",

constraint_values = [

"@platforms//os:freebsd",

"@platforms//cpu:x86_64",

],

visibility = ["//visibility:public"],

)

# Features that are default ON are handled differently below.

#

config_setting(

name = "no_gcp_support",

define_values = {"no_gcp_support": "true"},

visibility = ["//visibility:public"],

)

# Experimental features

config_setting(

name = "stackdriver_support",

define_values = {"stackdriver_support": "true"},

visibility = ["//visibility:public"],

)

config_setting(

name = "with_xla_support",

define_values = {"with_xla_support": "true"},

visibility = ["//visibility:public"],

)

# By default, XLA GPU is compiled into tensorflow when building with

# --config=cuda even when `with_xla_support` is false. The config setting

# here allows us to override the behavior if needed.

config_setting(

name = "no_xla_deps_in_cuda",

define_values = {"no_xla_deps_in_cuda": "true"},

visibility = ["//visibility:public"],

)

# Crosses between framework_shared_object and a bunch of other configurations

# due to limitations in nested select() statements.

config_setting(

name = "framework_shared_object",

define_values = {"framework_shared_object": "true"},

visibility = ["//visibility:public"],

)

config_setting(

name = "macos_x86_64_with_framework_shared_object",

define_values = {

"framework_shared_object": "true",

},

values = {

"apple_platform_type": "macos",

"cpu": "darwin",

},

visibility = ["//visibility:public"],

)

config_setting(

name = "macos_arm64_with_framework_shared_object",

define_values = {

"framework_shared_object": "true",

},

values = {

"apple_platform_type": "macos",

"cpu": "darwin_arm64",

},

visibility = ["//visibility:public"],

)

selects.config_setting_group(

name = "macos_with_framework_shared_object",

match_any = [

":macos_x86_64_with_framework_shared_object",

":macos_arm64_with_framework_shared_object",

],

)

# Config setting that is satisfied when TensorFlow is being built with CUDA

# support through e.g. `--config=cuda` (or `--config=cuda_clang` in OSS).

alias(

name = "is_cuda_enabled",

actual = if_oss(

"@local_config_cuda//:is_cuda_enabled",

"@local_config_cuda//cuda:using_config_cuda",

),

)

# Config setting that is satisfied when CUDA device code should be compiled

# with clang. It does not imply that CUDA support has been enabled.

alias(

name = "is_cuda_compiler_clang",

actual = if_oss(

"@local_config_cuda//:is_cuda_compiler_clang",

"@local_config_cuda//cuda:TRUE",

),

)

selects.config_setting_group(

name = "is_cuda_clang",

match_all = [

":is_cuda_enabled",

":is_cuda_compiler_clang",

],

)

# Config setting that is satisfied when CUDA device code should be compiled

# with nvcc. It does not imply that CUDA support has been enabled.

alias(

name = "is_cuda_compiler_nvcc",

actual = if_oss(

"@local_config_cuda//:is_cuda_compiler_nvcc",

"@local_config_cuda//cuda:FALSE",

),

)

selects.config_setting_group(

name = "is_cuda_nvcc",

match_all = [

":is_cuda_enabled",

":is_cuda_compiler_nvcc",

],

)

# Config setting that is satisfied when building with --config=cuda in OSS.

selects.config_setting_group(

name = "is_cuda_enabled_and_oss",

match_all = [

":is_cuda_enabled",

":oss",

],

)

# Config setting that is satisfied when building with --config=cuda for Windows

selects.config_setting_group(

name = "is_cuda_enabled_and_windows",

match_all = [

":is_cuda_enabled",

":windows",

],

)

# Config setting to use in select()s to distinguish open source build from

# google internal build on configurable attributes.

#

# For non-configurable distinction between OSS and Google builds, see

# `if_oss()` and `if_google()` macros in tensorflow.bzl.

config_setting(

name = "oss",

flag_values = {":oss_setting": "True"},

visibility = ["//visibility:public"],

)

# Non-configurable setting to indicate open source build.

bool_setting(

name = "oss_setting",

build_setting_default = if_oss(True, False),

visibility = ["//visibility:private"],

)

# Setting to use when loading kernels dynamically

config_setting(

name = "dynamic_loaded_kernels",

define_values = {

"dynamic_loaded_kernels": "true",

"framework_shared_object": "true",

},

visibility = ["//visibility:public"],

)

config_setting(

name = "using_rocm_hipcc",

define_values = {"using_rocm_hipcc": "true"},

)

config_setting(

name = "override_eigen_strong_inline",

define_values = {"override_eigen_strong_inline": "true"},

visibility = ["//visibility:public"],

)

# This flag specifies whether TensorFlow 2.0 API should be built instead

# of 1.* API. Note that TensorFlow 2.0 API is currently under development.

config_setting(

name = "api_version_2",

define_values = {"tf_api_version": "2"},

visibility = ["//visibility:public"],

)

# This flag specifies the file prefixes to strip from the file imports

# when generating python apis.

config_setting(

name = "litert_prefixes",

define_values = {"litert_prefixes": "external/org_tensorflow/"},

visibility = ["//visibility:public"],

)

# This flag disables generating tensorflow.lite.python under LiteRT repo.

# Avoid using flag for creating tflite wheels as tensorflow/lite is not yet fully split from tf.

config_setting(

name = "disable_tf_lite_py",

define_values = {"disable_tf_lite_py": "true"},

visibility = ["//visibility:public"],

)

# This flag enables experimental MLIR support.

config_setting(

name = "with_mlir_support",

define_values = {"with_mlir_support": "true"},

visibility = ["//visibility:public"],

)

# This flag forcibly enables experimental MLIR bridge support.

config_setting(

name = "enable_mlir_bridge",

define_values = {"enable_mlir_bridge": "true"},

visibility = ["//visibility:public"],

)

# This flag forcibly disables experimental MLIR bridge support.

config_setting(

name = "disable_mlir_bridge",

define_values = {"enable_mlir_bridge": "false"},

visibility = ["//visibility:public"],

)

# This flag enables experimental TPU support

bool_flag(

name = "enable_tpu_support",

build_setting_default = False,

)

config_setting(

name = "with_tpu_support_define",

define_values = {"with_tpu_support": "true"},

visibility = ["//visibility:public"],

)

config_setting(

name = "with_tpu_support_flag",

flag_values = {":enable_tpu_support": "True"},

visibility = ["//visibility:public"],

)

selects.config_setting_group(

name = "with_tpu_support",

match_any = [

":with_tpu_support_define",

":with_tpu_support_flag",

],

visibility = ["//visibility:public"],

)

# Specifies via a config setting if this is a mobile build or not, makes

# it easier to combine settings later.

selects.config_setting_group(

name = "mobile",

match_any = [

":android",

":chromiumos",

# TODO(jakeharmon8): Move all platform configs to TSL with fuchsia (non-blocking)

":fuchsia",

":emscripten",

":ios",

],

visibility = ["//visibility:public"],

)

config_setting(

name = "lite_protos_legacy",

define_values = {"TENSORFLOW_PROTOS": "lite"},

visibility = ["//visibility:public"],

)

config_setting(

name = "full_protos",

define_values = {"TENSORFLOW_PROTOS": "full"},

visibility = ["//visibility:public"],

)

selects.config_setting_group(

name = "lite_protos",

match_any = [":lite_protos_legacy"],

visibility = if_google(["//visibility:public"]),

)

selects.config_setting_group(

name = "mobile_lite_protos",

match_all = [

":lite_protos",

":mobile",

],

visibility = ["//visibility:public"],

)

selects.config_setting_group(

name = "mobile_full_protos",

match_all = [

":full_protos",

":mobile",

],

visibility = ["//visibility:public"],

)

# copybara:uncomment_begin(google-only)

# config_setting(

# name = "portable_proto_force_third_party",

# define_values = {"PORTABLE_PROTO_TRANSITION_MODE": "third_party"},

# visibility = ["//visibility:public"],

# )

# copybara:uncomment_end

# 'enable_registration_v2' opts-in to a different implementation of op and

# kernel registration - REGISTER_OP, REGISTER_KERNEL_BUILDER, etc.

#

# This setting is currently experimental. The 'v2' implementation does _not_

# correspond to a particular, finalized design; rather, it relates to

# developing one.

#

# The current aim of the 'v2' implementation is to allow 'unused' ops and

# kernels to be discarded by the linker (to the benefit of binary size).

bool_flag(

name = "enable_registration_v2",

build_setting_default = False,

visibility = ["//visibility:public"],

)

config_setting(

name = "registration_v1",

flag_values = {":enable_registration_v2": "False"},

visibility = ["//visibility:public"],

)

config_setting(

name = "registration_v2",

flag_values = {":enable_registration_v2": "True"},

visibility = ["//visibility:public"],

)