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

Cleanup keras build file for wrappers.

Browse Source 
PiperOrigin-RevId: 288806359
Change-Id: Ie3efa5f978c9a69700ac4758cceb066f2ba54657
This commit is contained in:
Scott Zhu 2020-01-08 17:14:57 -08:00 committed by TensorFlower Gardener
parent 9abca16f96
commit 65c7a019af
2 changed files with 42 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tensorflow/python/keras/BUILD
View File

@ -28,8 +28,6 @@ py_library(
"utils/multi_gpu_utils.py",
"utils/np_utils.py",
"utils/vis_utils.py",
"wrappers/__init__.py",
"wrappers/scikit_learn.py",
],
srcs_version = "PY2AND3",
visibility = ["//visibility:public"],
@ -45,6 +43,7 @@ py_library(
"//tensorflow/python/keras/mixed_precision/experimental:mixed_precision_experimental",
"//tensorflow/python/keras/optimizer_v2",
"//tensorflow/python/keras/premade",
"//tensorflow/python/keras/wrappers",
"//tensorflow/python/saved_model",
],
)
@ -1275,20 +1274,6 @@ tf_py_test(
],
)
tf_py_test(
name = "scikit_learn_test",
size = "small",
srcs = ["wrappers/scikit_learn_test.py"],
python_version = "PY3",
tags = ["notsan"],
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
name = "data_utils_test",
size = "medium",

41
tensorflow/python/keras/wrappers/BUILD Normal file
View File

@ -0,0 +1,41 @@
# Description:
# Contains the Keras wrapper API (internal TensorFlow version).
load("//tensorflow:tensorflow.bzl", "tf_py_test")
package(
default_visibility = ["//visibility:public"],
licenses = ["notice"], # Apache 2.0
)
exports_files(["LICENSE"])
py_library(
name = "wrappers",
srcs = [
"__init__.py",
"scikit_learn.py",
],
srcs_version = "PY2AND3",
deps = [
"//tensorflow/python:util",
"//tensorflow/python/keras:engine",
"//tensorflow/python/keras:generic_utils",
"//tensorflow/python/keras:losses",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "scikit_learn_test",
size = "small",
srcs = ["scikit_learn_test.py"],
python_version = "PY3",
tags = ["notsan"],
deps = [
":wrappers",
"//tensorflow/python:client_testlib",
"//tensorflow/python:extra_py_tests_deps",
"//third_party/py/numpy",
],
)