Bugfixes to TF to allow tensorflow_serving to build.

PiperOrigin-RevId: 300195183
Change-Id: I20c40f96400013d507053accb7bf19f1f9083330
This commit is contained in:
Eugene Brevdo 2020-03-10 15:47:24 -07:00 committed by TensorFlower Gardener
parent 658295136d
commit 003659ffcc
2 changed files with 15 additions and 5 deletions

View File

@ -1,3 +1,5 @@
load("//tensorflow:tensorflow.bzl", "tf_external_workspace_visible")
package(
licenses = ["notice"], # Apache 2.0
)
@ -7,7 +9,9 @@ cc_library(
srcs = ["profiler_service_impl.cc"],
hdrs = ["profiler_service_impl.h"],
features = ["-layering_check"],
visibility = ["//tensorflow_serving/model_servers:__pkg__"],
visibility = tf_external_workspace_visible(
["//tensorflow_serving/model_servers:__pkg__"],
),
deps = [
"//tensorflow:grpc++",
"//tensorflow/core:framework",

View File

@ -2213,6 +2213,15 @@ def tf_py_test(
if grpc_enabled:
deps = deps + tf_additional_grpc_deps_py()
# NOTE(ebrevdo): This is a workaround for depset() not being able to tell
# the difference between 'dep' and 'clean_dep(dep)'.
for to_add in [
"//tensorflow/python:extra_py_tests_deps",
"//tensorflow/python:gradient_checker",
]:
if to_add not in deps:
deps.append(clean_dep(to_add))
# Python version placeholder
kwargs.setdefault("srcs_version", "PY2AND3")
py_test(
@ -2228,10 +2237,7 @@ def tf_py_test(
tags = tags,
visibility = [clean_dep("//tensorflow:internal")] +
additional_visibility,
deps = depset([
clean_dep("//tensorflow/python:extra_py_tests_deps"),
clean_dep("//tensorflow/python:gradient_checker"),
] + deps + xla_test_true_list),
deps = depset(deps + xla_test_true_list),
**kwargs
)