# Description:

# Python support for TensorFlow.

#

# Public targets:

# ":platform" - Low-level and platform-specific Python code.

visibility = [

"//engedu/ml/tf_from_scratch:__pkg__",

"//tensorflow:internal",

"//tensorflow/contrib/lite/toco/python:__pkg__",

"//tensorflow_models:__subpackages__",

# TODO(aselle): to pass open source test.

"//bazel_pip/tensorflow/contrib/lite/toco/python:__pkg__",

]

package(default_visibility = visibility)

licenses(["notice"]) # Apache 2.0

exports_files(["LICENSE"])

load("//tensorflow:tensorflow.bzl", "if_not_windows")

load("//tensorflow:tensorflow.bzl", "tf_cuda_library")

load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py")

load("//tensorflow:tensorflow.bzl", "py_test")

load("//tensorflow:tensorflow.bzl", "tf_py_test")

load("//tensorflow:tensorflow.bzl", "py_tests")

load("//tensorflow:tensorflow.bzl", "tf_py_build_info_genrule")

load("//tensorflow:tensorflow.bzl", "tf_py_wrap_cc")

load("//tensorflow:tensorflow.bzl", "tf_cc_shared_object")

load("//tensorflow:tensorflow.bzl", "tf_native_cc_binary")

load("//tensorflow:tensorflow.bzl", "tf_custom_op_library_additional_deps_impl")

load("//tensorflow:tensorflow.bzl", "cuda_py_test")

load("//tensorflow:tensorflow.bzl", "cuda_py_tests")

load("//tensorflow/core:platform/default/build_config.bzl", "pyx_library")

load("//tensorflow/core:platform/default/build_config.bzl", "tf_proto_library")

load("//tensorflow/core:platform/default/build_config.bzl", "tf_proto_library_py")

load("//tensorflow/core:platform/default/build_config.bzl", "tf_additional_lib_deps")

load("//tensorflow/core:platform/default/build_config.bzl", "tf_additional_all_protos")

load("//tensorflow/core:platform/default/build_config.bzl", "tf_protos_grappler")

load("//tensorflow/core:platform/default/build_config_root.bzl", "tf_additional_plugin_deps")

load("//tensorflow/python:build_defs.bzl", "tf_gen_op_wrapper_private_py")

load("//tensorflow/core:platform/default/build_config_root.bzl", "tf_additional_verbs_deps")

load("//tensorflow/core:platform/default/build_config_root.bzl", "tf_additional_mpi_deps")

load("//tensorflow/core:platform/default/build_config_root.bzl", "tf_additional_gdr_deps")

load("//tensorflow/core:platform/default/build_config_root.bzl", "if_static")

load(

"//third_party/ngraph:build_defs.bzl",

"if_ngraph",

)

py_library(

name = "python",

srcs = ["__init__.py"],

srcs_version = "PY2AND3",

visibility = [

"//tensorflow:__pkg__",

"//tensorflow/compiler/aot/tests:__pkg__", # TODO(b/34059704): remove when fixed

"//tensorflow/contrib/learn:__pkg__", # TODO(b/34059704): remove when fixed

"//tensorflow/contrib/learn/python/learn/datasets:__pkg__", # TODO(b/34059704): remove when fixed

"//tensorflow/contrib/lite/toco/python:__pkg__", # TODO(b/34059704): remove when fixed

"//tensorflow/python/debug:__pkg__", # TODO(b/34059704): remove when fixed

"//tensorflow/python/tools:__pkg__", # TODO(b/34059704): remove when fixed

"//tensorflow/tools/quantization:__pkg__", # TODO(b/34059704): remove when fixed

],

deps = [

":no_contrib",

"//tensorflow/contrib:contrib_py",

"//tensorflow/python/estimator:estimator_py",

],

)

py_library(

name = "no_contrib",

srcs = ["__init__.py"],

srcs_version = "PY2AND3",

visibility = [

"//tensorflow:__pkg__",

"//tensorflow/python/tools:__pkg__",

"//tensorflow/python/tools/api/generator:__pkg__",

"//tensorflow/tools/api/tests:__pkg__",

],

deps = [

":array_ops",

":bitwise_ops",

":boosted_trees_ops",

":check_ops",

":client",

":client_testlib",

":collective_ops",

":cond_v2",

":confusion_matrix",

":control_flow_ops",

":cudnn_rnn_ops_gen",

":errors",

":framework",

":framework_for_generated_wrappers",

":functional_ops",

":gradient_checker",

":graph_util",

":histogram_ops",

":image_ops",

":initializers_ns",

":io_ops",

":kernels",

":layers",

":lib",

":list_ops",

":manip_ops",

":math_ops",

":metrics",

":nn",

":ops",

":platform",

":pywrap_tensorflow",

":saver_test_utils",

":script_ops",

":session_ops",

":sets",

":sparse_ops",

":spectral_ops",

":spectral_ops_test_util",

":standard_ops",

":state_ops",

":string_ops",

":subscribe",

":summary",

":tensor_array_ops",

":test_ops", # TODO: Break testing code out into separate rule.

":tf_cluster",

":tf_item",

":tf_optimizer",

":training",

":util",

":weights_broadcast_ops",

":while_v2",

"//tensorflow/core:protos_all_py",

"//tensorflow/python/compat",

"//tensorflow/python/data",

"//tensorflow/python/distribute:estimator_training",

"//tensorflow/python/feature_column:feature_column_py",

"//tensorflow/python/keras",

"//tensorflow/python/ops/distributions",

"//tensorflow/python/ops/linalg",

"//tensorflow/python/ops/losses",

"//tensorflow/python/ops/parallel_for",

"//tensorflow/python/profiler",

"//tensorflow/python/saved_model",

"//tensorflow/python/tools:component_api_helper",

"//tensorflow/python/tools/api/generator:create_python_api",

"//third_party/py/numpy",

],

)

tf_py_build_info_genrule()

py_library(

name = "platform",

srcs = glob(

[

"platform/*.py",

],

exclude = [

"**/*test.py",

"**/benchmark.py", # In platform_benchmark.

],

) + ["platform/build_info.py"],

srcs_version = "PY2AND3",

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

deps = [

":lib",

":pywrap_tensorflow",

":util",

"//tensorflow/core:protos_all_py",

"@absl_py//absl/flags",

"@six_archive//:six",

],

)

py_library(

name = "platform_benchmark",

srcs = ["platform/benchmark.py"],

srcs_version = "PY2AND3",

deps = [

":client",

":platform",

"@six_archive//:six",

],

)

py_library(

name = "platform_test",

srcs = ["platform/googletest.py"],

srcs_version = "PY2AND3",

deps = [":platform_benchmark"],

)

tf_py_test(

name = "resource_loader_test",

size = "small",

srcs = ["platform/resource_loader_test.py"],

additional_deps = [

":platform",

":platform_test",

],

)

tf_py_test(

name = "flags_test",

size = "small",

srcs = ["platform/flags_test.py"],

additional_deps = [

":client_testlib",

":platform",

],

)

tf_py_test(

name = "stacktrace_handler_test",

size = "small",

srcs = ["platform/stacktrace_handler_test.py"],

additional_deps = [

":client_testlib",

":platform",

],

tags = ["no_windows"],

)

tf_py_test(

name = "app_test",

size = "small",

srcs = ["platform/app_test.py"],

additional_deps = [":platform"],

tags = ["notap"],

)

cc_library(

name = "cost_analyzer_lib",

srcs = ["grappler/cost_analyzer.cc"],

hdrs = ["grappler/cost_analyzer.h"],

deps = [

"//tensorflow/core:lib",

"//tensorflow/core:protos_all_cc",

"//tensorflow/core/grappler:grappler_item",

"//tensorflow/core/grappler/clusters:cluster",

"//tensorflow/core/grappler/costs:analytical_cost_estimator",

"//tensorflow/core/grappler/costs:cost_estimator",

"//tensorflow/core/grappler/costs:measuring_cost_estimator",

"//tensorflow/core/grappler/costs:utils",

] + tf_protos_grappler(),

)

cc_library(

name = "model_analyzer_lib",

srcs = ["grappler/model_analyzer.cc"],

hdrs = ["grappler/model_analyzer.h"],

deps = [

"//tensorflow/core:framework",

"//tensorflow/core:lib",

"//tensorflow/core:protos_all_cc",

"//tensorflow/core/grappler:grappler_item",

"//tensorflow/core/grappler/costs:graph_properties",

],

)

cc_library(

name = "numpy_lib",

srcs = ["lib/core/numpy.cc"],

hdrs = ["lib/core/numpy.h"],

deps = [

"//tensorflow/core:framework",

"//tensorflow/core:lib",

"//third_party/py/numpy:headers",

"//third_party/python_runtime:headers",

],

)

cc_library(

name = "bfloat16_lib",

srcs = ["lib/core/bfloat16.cc"],

hdrs = ["lib/core/bfloat16.h"],

deps = [

":numpy_lib",

":safe_ptr",

"//tensorflow/core:framework",

"//tensorflow/core:lib",

"//third_party/python_runtime:headers",

],

)

cc_library(

name = "ndarray_tensor_bridge",

srcs = ["lib/core/ndarray_tensor_bridge.cc"],

hdrs = ["lib/core/ndarray_tensor_bridge.h"],

visibility = visibility + [

"//learning/deepmind/courier:__subpackages__",

],

deps = [

":bfloat16_lib",

":numpy_lib",

"//tensorflow/c:c_api",

"//tensorflow/core:lib",

"//tensorflow/core:protos_all_cc",

],

)

cc_library(

name = "py_exception_registry",

srcs = ["lib/core/py_exception_registry.cc"],

hdrs = ["lib/core/py_exception_registry.h"],

deps = [

"//tensorflow/c:c_api",

"//tensorflow/core:lib",

"//third_party/python_runtime:headers",

],

)

cc_library(

name = "kernel_registry",

srcs = ["util/kernel_registry.cc"],

hdrs = ["util/kernel_registry.h"],

deps = [

"//tensorflow/core:framework",

"//tensorflow/core:lib",

"//tensorflow/core:protos_all_cc",

],

)

cc_library(

name = "cpp_python_util",

srcs = ["util/util.cc"],

hdrs = ["util/util.h"],

deps = [

":safe_ptr",

"//tensorflow/core:lib",

"//tensorflow/core:lib_internal",

"//third_party/python_runtime:headers",

"@com_google_absl//absl/memory",

],

)

cc_library(

name = "py_func_lib",

srcs = ["lib/core/py_func.cc"],

hdrs = ["lib/core/py_func.h"],

deps = [

":ndarray_tensor_bridge",

":numpy_lib",

":py_util",

":safe_ptr",

"//tensorflow/c:tf_status_helper",

"//tensorflow/c/eager:c_api",

"//tensorflow/c/eager:c_api_internal",

"//tensorflow/core:framework",

"//tensorflow/core:lib",

"//tensorflow/core:protos_all_cc",

"//tensorflow/core:script_ops_op_lib",

"//tensorflow/python/eager:pywrap_tfe_lib",

"//third_party/py/numpy:headers",

"//third_party/python_runtime:headers",

],

)

cc_library(

name = "safe_ptr",

srcs = ["lib/core/safe_ptr.cc"],

hdrs = ["lib/core/safe_ptr.h"],

deps = [

"//tensorflow/c:c_api",

"//tensorflow/c/eager:c_api",

"//third_party/python_runtime:headers",

],

)

cc_library(

name = "ndarray_tensor",

srcs = ["lib/core/ndarray_tensor.cc"],

hdrs = ["lib/core/ndarray_tensor.h"],

visibility = visibility + [

"//learning/deepmind/courier:__subpackages__",

],

deps = [

":bfloat16_lib",

":ndarray_tensor_bridge",

":numpy_lib",

":safe_ptr",

"//tensorflow/c:c_api",

"//tensorflow/c:tf_status_helper",

"//tensorflow/core:framework",

"//tensorflow/core:lib",

],

)

cc_library(

name = "py_seq_tensor",

srcs = ["lib/core/py_seq_tensor.cc"],

hdrs = ["lib/core/py_seq_tensor.h"],

deps = [

":numpy_lib",

":py_util",

":safe_ptr",

"//tensorflow/core:framework",

"//tensorflow/core:lib",

"//third_party/python_runtime:headers",

],

)

cc_library(

name = "py_util",

srcs = ["lib/core/py_util.cc"],

hdrs = ["lib/core/py_util.h"],

deps = [

"//tensorflow/core:lib",

"//tensorflow/core:script_ops_op_lib",

"//third_party/python_runtime:headers",

],

)

cc_library(

name = "py_record_reader_lib",

srcs = ["lib/io/py_record_reader.cc"],

hdrs = ["lib/io/py_record_reader.h"],

deps = [

"//tensorflow/c:c_api",

"//tensorflow/c:tf_status_helper",

"//tensorflow/core:lib",

"//tensorflow/core:lib_internal",

],

)

cc_library(

name = "py_record_writer_lib",

srcs = ["lib/io/py_record_writer.cc"],

hdrs = ["lib/io/py_record_writer.h"],

deps = [

"//tensorflow/c:c_api",

"//tensorflow/c:tf_status_helper",

"//tensorflow/core:lib",

"//tensorflow/core:lib_internal",

],

)

tf_cc_shared_object(

name = "framework/test_file_system.so",

srcs = ["framework/test_file_system.cc"],

copts = if_not_windows(["-Wno-sign-compare"]),

linkopts = select({

"//conditions:default": [

"-lm",

],

"//tensorflow:darwin": [],

"//tensorflow:windows": [],

}),

deps = [

"//tensorflow/core:framework_headers_lib",

"//third_party/eigen3",

"@protobuf_archive//:protobuf_headers",

],

)

py_test(

name = "file_system_test",

size = "small",

srcs = ["framework/file_system_test.py"],

data = [":framework/test_file_system.so"],

main = "framework/file_system_test.py",

srcs_version = "PY2AND3",

tags = [

"no_pip", # Path issues due to test environment

"no_windows",

"notap",

],

deps = [

":client_testlib",

":data_flow_ops",

":framework",

":framework_for_generated_wrappers",

":io_ops",

":platform",

":util",

],

)

py_test(

name = "decorator_utils_test",

srcs = ["util/decorator_utils_test.py"],

srcs_version = "PY2AND3",

deps = [

":client_testlib",

":platform",

":util",

],

)

py_test(

name = "tf_export_test",

srcs = ["util/tf_export_test.py"],

srcs_version = "PY2AND3",

deps = [

":client_testlib",

":platform",

":util",

],

)

py_test(

name = "deprecation_test",

srcs = ["util/deprecation_test.py"],

srcs_version = "PY2AND3",

deps = [

":client_testlib",

":platform",

":util",

],

)

py_test(

name = "keyword_args_test",

srcs = ["util/keyword_args_test.py"],

srcs_version = "PY2AND3",

deps = [

":client_testlib",

":util",

],

)

cc_library(

name = "python_op_gen",

srcs = [

"framework/python_op_gen.cc",

"framework/python_op_gen_internal.cc",

],

hdrs = [

"framework/python_op_gen.h",

"framework/python_op_gen_internal.h",

],

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

deps = [

"//tensorflow/core:framework",

"//tensorflow/core:lib",

"//tensorflow/core:lib_internal",

"//tensorflow/core:op_gen_lib",

"//tensorflow/core:proto_text",

"//tensorflow/core:protos_all_cc",

],

alwayslink = 1,

)

cc_library(

name = "python_op_gen_main",

srcs = ["framework/python_op_gen_main.cc"],

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

deps = [

":python_op_gen",

"//tensorflow/core:framework",

"//tensorflow/core:lib",

"//tensorflow/core:lib_internal",

"//tensorflow/core:op_gen_lib",

"//tensorflow/core:protos_all_cc",

],

)

py_library(

name = "framework_for_generated_wrappers",

srcs_version = "PY2AND3",

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

deps = [

":constant_op",

":device",

":dtypes",

":framework_ops",

":function",

":op_def_library",

":op_def_registry",

":registry",

":tensor_shape",

":versions",

],

)

# What is needed for tf_gen_op_wrapper_py. This is the same as

# "framework_for_generated_wrappers" minus the "function" dep. This is to avoid

# circular dependencies, as "function" uses generated op wrappers.

py_library(

name = "framework_for_generated_wrappers_v2",

srcs_version = "PY2AND3",

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

deps = [

":constant_op",

":device",

":dtypes",

":framework_ops",

":op_def_library",

":op_def_registry",

":registry",

":tensor_shape",

":versions",

"//tensorflow/python/eager:context",

"//tensorflow/python/eager:core",

"//tensorflow/python/eager:execute",

],

)

py_library(

name = "subscribe",

srcs = ["framework/subscribe.py"],

srcs_version = "PY2AND3",

deps = [

":array_ops",

":framework_ops",

":platform",

":variables",

],

)

py_library(

name = "framework",

srcs = [

"framework/framework_lib.py",

"framework/graph_io.py",

"framework/importer.py",

"framework/load_library.py",

"framework/meta_graph.py",

],

srcs_version = "PY2AND3",

deps = [

":common_shapes",

":cpp_shape_inference_proto_py",

":errors",

":framework_fast_tensor_util",

":framework_for_generated_wrappers",

":function",

":graph_util",

":lib",

":platform",

":pywrap_tensorflow",

":random_seed",

":sparse_tensor",

":tensor_spec",

":tensor_util",

":util",

"//tensorflow/python/eager:context",

"//third_party/py/numpy",

"@six_archive//:six",

],

)

py_library(

name = "c_api_util",

srcs = ["framework/c_api_util.py"],

srcs_version = "PY2AND3",

deps = [

":pywrap_tensorflow",

"//tensorflow/core:protos_all_py",

],

)

py_library(

name = "common_shapes",

srcs = ["framework/common_shapes.py"],

srcs_version = "PY2AND3",

deps = [

":cpp_shape_inference_proto_py",

":errors",

":framework_ops",

":pywrap_tensorflow",

":tensor_shape",

":tensor_util",

"//tensorflow/core:protos_all_py",

],

)

py_library(

name = "constant_op",

srcs = ["framework/constant_op.py"],

srcs_version = "PY2AND3",

deps = [

":dtypes",

":framework_ops",

":tensor_shape",

"//tensorflow/core:protos_all_py",

"//tensorflow/python/eager:execute",

],

)

py_library(

name = "device",

srcs = ["framework/device.py"],

srcs_version = "PY2AND3",

)

py_library(

name = "dtypes",

srcs = ["framework/dtypes.py"],

srcs_version = "PY2AND3",

deps = [

":pywrap_tensorflow",

"//tensorflow/core:protos_all_py",

],

)

py_library(

name = "errors",

srcs = [

"framework/errors.py",

"framework/errors_impl.py",

],

srcs_version = "PY2AND3",

deps = [

":c_api_util",

":util",

],

)

py_library(

name = "error_interpolation",

srcs = [

"framework/error_interpolation.py",

],

srcs_version = "PY2AND3",

deps = [

":util",

],

)

py_library(

name = "function",

srcs = ["framework/function.py"],

srcs_version = "PY2AND3",

deps = [

":array_ops",

":dtypes",

":framework_ops",

":graph_to_function_def",

":op_def_registry",

":util",

":variable_scope",

"//tensorflow/core:protos_all_py",

"//tensorflow/python/eager:context",

],

)

py_library(

name = "graph_to_function_def",

srcs = ["framework/graph_to_function_def.py"],

srcs_version = "PY2AND3",

deps = [

":op_def_registry",

"//tensorflow/core:protos_all_py",

],

)

py_library(

name = "function_def_to_graph",

srcs = ["framework/function_def_to_graph.py"],

srcs_version = "PY2AND3",

deps = [

":framework",

":framework_ops",

":function",

":tensor_shape",

":versions",

"//tensorflow/core:protos_all_py",

],

)

py_test(

name = "function_def_to_graph_test",

size = "small",

srcs = ["framework/function_def_to_graph_test.py"],

srcs_version = "PY2AND3",

tags = ["no_pip"],

deps = [

":array_ops",

":client_testlib",

":constant_op",

":dtypes",

":framework_ops",

":function",

":function_def_to_graph",

":graph_to_function_def",

":math_ops",

":test_ops",

],

)

py_library(

name = "graph_util",

srcs = [

"framework/graph_util.py",

"framework/graph_util_impl.py",

],

srcs_version = "PY2AND3",

deps = [

":dtypes",

":framework_ops",

":platform",

":tensor_util",

"//tensorflow/core:protos_all_py",

],

)

py_library(

name = "kernels",

srcs = [

"framework/kernels.py",

],

srcs_version = "PY2AND3",

deps = [

":pywrap_tensorflow",

":util",

"//tensorflow/core:protos_all_py",

],

)

py_library(

name = "op_def_library",

srcs = ["framework/op_def_library.py"],

srcs_version = "PY2AND3",

deps = [

":dtypes",

":framework_ops",

":platform",

":tensor_shape",

":util",

"//tensorflow/core:protos_all_py",

"@six_archive//:six",

],

)

py_library(

name = "op_def_registry",

srcs = ["framework/op_def_registry.py"],

srcs_version = "PY2AND3",

deps = [

"//tensorflow/core:protos_all_py",

],

)

py_library(

name = "framework_ops", # "ops" is already the name of a deprecated target

srcs = ["framework/ops.py"],

srcs_version = "PY2AND3",

deps = [

":c_api_util",

":control_flow_util",

":cpp_shape_inference_proto_py",

":device",

":dtypes",

":error_interpolation",

":op_def_registry",

":platform",

":registry",

":tensor_shape",

":traceable_stack",

":util",

":versions",

"//tensorflow/core:protos_all_py",

"//tensorflow/python/eager:context",

"//tensorflow/python/eager:core",

"//tensorflow/python/eager:tape",

"@six_archive//:six",

],

)

py_library(

name = "func_graph",

srcs = ["framework/func_graph.py"],

srcs_version = "PY2AND3",

deps = [

":auto_control_deps",

":framework_ops",

"//tensorflow/python/autograph",

"//tensorflow/python/eager:context",

"//tensorflow/python/eager:graph_only_ops",

"//tensorflow/python/eager:tape",

],

)

py_library(

name = "auto_control_deps",

srcs = ["framework/auto_control_deps.py"],

srcs_version = "PY2AND3",

deps = [

":control_flow_ops",

":framework_ops",

":util",

],

)

py_test(

name = "auto_control_deps_test",

size = "small",

srcs = ["framework/auto_control_deps_test.py"],

srcs_version = "PY2AND3",

deps = [

":auto_control_deps",

":client_testlib",

],

)

py_library(

name = "random_seed",

srcs = ["framework/random_seed.py"],

srcs_version = "PY2AND3",

deps = [

":framework_ops",

],

)

py_library(

name = "registry",

srcs = ["framework/registry.py"],

srcs_version = "PY2AND3",

deps = [

":platform",

":util",

],

)

py_library(

name = "smart_cond",

srcs = ["framework/smart_cond.py"],

srcs_version = "PY2AND3",

deps = [

":control_flow_ops",

":tensor_util",

],

)

py_test(

name = "smart_cond_test",

size = "small",

srcs = ["framework/smart_cond_test.py"],

srcs_version = "PY2AND3",

deps = [

":client_testlib",

":constant_op",

":framework_ops",

":math_ops",

":session",

":smart_cond",

],

)

py_library(

name = "sparse_tensor",

srcs = ["framework/sparse_tensor.py"],

srcs_version = "PY2AND3",

deps = [

":dtypes",

":framework_ops",

":tensor_util",

],

)

# This target is maintained separately from :util to provide separate visibility

# for legacy users who were granted visibility when the functions were private

# members of ops.Graph.

py_library(

name = "tf_stack",

srcs = ["util/tf_stack.py"],

srcs_version = "PY2AND3",

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

deps = [],

)

py_library(

name = "tensor_shape",

srcs = ["framework/tensor_shape.py"],

srcs_version = "PY2AND3",

deps = [

":dtypes",

":util",

"//tensorflow/core:protos_all_py",

],

)

py_library(

name = "tensor_spec",

srcs = ["framework/tensor_spec.py"],

srcs_version = "PY2AND3",

deps = [

":common_shapes",

":dtypes",

":tensor_shape",

"//third_party/py/numpy",

],

)

py_library(

name = "tensor_util",

srcs = ["framework/tensor_util.py"],

srcs_version = "PY2AND3",

deps = [

":tensor_shape",

":util",

"//tensorflow/core:protos_all_py",

],

)