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

Create no-src target for expanding visibility of TF deps.

Browse Source 
PiperOrigin-RevId: 322278685
Change-Id: Id1295a5de3029ca874cd3908e3f09b5fef88bea7
This commit is contained in:
Revan Sopher 2020-07-20 20:03:36 -07:00 committed by TensorFlower Gardener
parent 322791ebc8
commit 8dbdd0803f
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/tpu/BUILD
View File

@ -560,6 +560,13 @@ tpu_py_test(
], ],
) )
# NOTE this target should only be depended on by the tpu_test_wrapper macro.
py_library(
name = "tpu_test_deps",
visibility = ["//visibility:public"],
deps = ["//tensorflow/python:client_testlib"],
)
tf_proto_library( tf_proto_library(
name = "tensor_tracer_proto", name = "tensor_tracer_proto",
srcs = ["tensor_tracer.proto"], srcs = ["tensor_tracer.proto"],

2
tensorflow/python/tpu/tpu_test_wrapper.bzl
View File

@ -57,7 +57,7 @@ def get_kwargs_for_wrapping(
kwargs["python_version"] = kwargs.get("python_version", "PY3") kwargs["python_version"] = kwargs.get("python_version", "PY3")
kwargs["srcs"] = [wrapper_src] + kwargs["srcs"] kwargs["srcs"] = [wrapper_src] + kwargs["srcs"]
kwargs["deps"] = depset( kwargs["deps"] = depset(
["//tensorflow/python:client_testlib"], ["//tensorflow/python/tpu:tpu_test_deps"],
transitive = [deps], transitive = [deps],
) )
kwargs["main"] = wrapper_src kwargs["main"] = wrapper_src