experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

allow python tracer(using embedded python internally) to be linked to pywrap_tensorflow target by default for google platform.

Browse Source 
PiperOrigin-RevId: 314574561
Change-Id: I141ddcb5129abeedfff505b7a7c6ce43acb03bc1
This commit is contained in:
A. Unique TensorFlower 2020-06-03 11:52:16 -07:00 committed by TensorFlower Gardener
parent 33b3e6cd06
commit a4799e5246
3 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/platform/build_config_root.bzl
View File

@ -9,6 +9,7 @@ load(
_tf_additional_grpc_deps_py = "tf_additional_grpc_deps_py",
_tf_additional_license_deps = "tf_additional_license_deps",
_tf_additional_plugin_deps = "tf_additional_plugin_deps",
_tf_additional_profiler_deps = "tf_additional_profiler_deps",
_tf_additional_xla_deps_py = "tf_additional_xla_deps_py",
_tf_cuda_tests_tags = "tf_cuda_tests_tags",
_tf_exec_properties = "tf_exec_properties",
@ -23,6 +24,7 @@ register_extension_info = _register_extension_info
tf_additional_grpc_deps_py = _tf_additional_grpc_deps_py
tf_additional_license_deps = _tf_additional_license_deps
tf_additional_plugin_deps = _tf_additional_plugin_deps
tf_additional_profiler_deps = _tf_additional_profiler_deps
tf_additional_xla_deps_py = _tf_additional_xla_deps_py
tf_cuda_tests_tags = _tf_cuda_tests_tags
tf_exec_properties = _tf_exec_properties

3
tensorflow/core/platform/default/build_config_root.bzl
View File

@ -35,6 +35,9 @@ def tf_additional_plugin_deps():
"//conditions:default": [],
})
def tf_additional_profiler_deps():
return []
def tf_additional_xla_deps_py():
return []

5
tensorflow/python/BUILD
View File

@ -32,7 +32,7 @@ load("//tensorflow:tensorflow.bzl", "tf_py_test")
# buildifier: disable=same-origin-load
load("//tensorflow:tensorflow.bzl", "tf_py_build_info_genrule")
load("//tensorflow/core/platform:build_config.bzl", "pyx_library", "tf_additional_all_protos", "tf_additional_lib_deps", "tf_proto_library", "tf_proto_library_py", "tf_protos_grappler") # @unused
load("//tensorflow/core/platform:build_config_root.bzl", "if_static", "tf_additional_plugin_deps", "tf_additional_xla_deps_py")
load("//tensorflow/core/platform:build_config_root.bzl", "if_static", "tf_additional_plugin_deps", "tf_additional_profiler_deps", "tf_additional_xla_deps_py")
load("//tensorflow/python:build_defs.bzl", "tf_gen_op_wrapper_private_py")
load(
"//third_party/ngraph:build_defs.bzl",
@ -6012,7 +6012,8 @@ pywrap_tensorflow_macro(
"//tensorflow/core/util/tensor_bundle",
"//tensorflow/compiler/mlir/python:mlir",
] + (tf_additional_lib_deps() +
tf_additional_plugin_deps()) + if_ngraph([
tf_additional_plugin_deps() +
tf_additional_profiler_deps()) + if_ngraph([
"@ngraph_tf//:ngraph_tf",
]) + if_xla_available([
"//tensorflow/compiler/aot:tfcompile_lib",