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

Add explicit build dependency for kpl_test_utils.

Browse Source 
PiperOrigin-RevId: 357763129
Change-Id: Iecd1f3371f8adba03dcbbd441280e387a8d26a9f
This commit is contained in:
Scott Zhu 2021-02-16 11:18:59 -08:00 committed by TensorFlower Gardener
parent 22081f66ca
commit c3555ff1fc
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/keras/utils/BUILD
View File

@ -9,6 +9,7 @@ package(
default_visibility = [
"//tensorflow/python/feature_column:__pkg__",
"//tensorflow/python/keras:__subpackages__",
"//tensorflow/tools/pip_package:__pkg__",
],
licenses = ["notice"], # Apache 2.0
)
@ -40,7 +41,6 @@ py_library(
":control_flow_util",
":engine_utils",
":generic_utils",
":kpl_test_utils",
":layer_utils",
":multi_gpu_utils",
":np_utils",
@ -59,7 +59,10 @@ py_library(
name = "kpl_test_utils",
srcs = ["kpl_test_utils.py"],
srcs_version = "PY3",
deps = [],
deps = [
"//tensorflow/python/keras",
"//tensorflow/python/keras/layers/preprocessing:string_lookup",
],
)
py_library(

1
tensorflow/tools/pip_package/BUILD
View File

@ -135,6 +135,7 @@ COMMON_PIP_DEPS = [
"//tensorflow/python/keras/tests:model_subclassing_test_util",
"//tensorflow/python/keras/tests:model_architectures",
"//tensorflow/python/keras/utils:dataset_creator",
"//tensorflow/python/keras/utils:kpl_test_utils",
"//tensorflow/python/keras/benchmarks:keras_benchmark_lib_pip",
"//tensorflow/python/kernel_tests:cudnn_deterministic_base",
"//tensorflow/python/kernel_tests:bias_op_base",