Create no-src target for expanding visibility of TF deps.
PiperOrigin-RevId: 322278685 Change-Id: Id1295a5de3029ca874cd3908e3f09b5fef88bea7
This commit is contained in:
parent
322791ebc8
commit
8dbdd0803f
@ -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(
|
||||
name = "tensor_tracer_proto",
|
||||
srcs = ["tensor_tracer.proto"],
|
||||
|
@ -57,7 +57,7 @@ def get_kwargs_for_wrapping(
|
||||
kwargs["python_version"] = kwargs.get("python_version", "PY3")
|
||||
kwargs["srcs"] = [wrapper_src] + kwargs["srcs"]
|
||||
kwargs["deps"] = depset(
|
||||
["//tensorflow/python:client_testlib"],
|
||||
["//tensorflow/python/tpu:tpu_test_deps"],
|
||||
transitive = [deps],
|
||||
)
|
||||
kwargs["main"] = wrapper_src
|
||||
|
Loading…
Reference in New Issue
Block a user