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

Update build file for keras/preprocessing.

Browse Source 
PiperOrigin-RevId: 289360563
Change-Id: I494fd585d56adfb7928324a16cbb2ed67322820d
This commit is contained in:
Scott Zhu 2020-01-12 17:58:18 -08:00 committed by TensorFlower Gardener
parent 99a913907c
commit f6273c2a78
3 changed files with 99 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
tensorflow/python/keras/BUILD
View File

@ -18,10 +18,6 @@ py_library(
"estimator/__init__.py",
"keras_parameterized.py",
"ops.py",
"preprocessing/__init__.py",
"preprocessing/image.py",
"preprocessing/sequence.py",
"preprocessing/text.py",
"testing_utils.py",
],
srcs_version = "PY2AND3",
@ -37,6 +33,7 @@ py_library(
"//tensorflow/python/keras/mixed_precision/experimental:mixed_precision_experimental",
"//tensorflow/python/keras/optimizer_v2",
"//tensorflow/python/keras/premade",
"//tensorflow/python/keras/preprocessing",
"//tensorflow/python/keras/saving",
"//tensorflow/python/keras/utils",
"//tensorflow/python/keras/wrappers",
@ -1162,45 +1159,6 @@ cuda_py_test(
],
)
tf_py_test(
name = "image_test",
size = "medium",
srcs = ["preprocessing/image_test.py"],
python_version = "PY3",
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
name = "sequence_test",
size = "small",
srcs = ["preprocessing/sequence_test.py"],
python_version = "PY3",
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
name = "text_test",
size = "small",
srcs = ["preprocessing/text_test.py"],
python_version = "PY3",
deps = [
":keras",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
"@absl_py//absl/testing:parameterized",
],
)
tf_py_test(
name = "callbacks_test",
size = "medium",

90
tensorflow/python/keras/preprocessing/BUILD Normal file
View File

@ -0,0 +1,90 @@
# Description:
# Contains the Keras preprocessing layers (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 = "preprocessing",
srcs = [
"__init__.py",
],
deps = [
":image",
":sequence",
":text",
],
)
py_library(
name = "image",
srcs = [
"image.py",
],
deps = [
"//tensorflow/python:util",
"//tensorflow/python/keras:backend",
"//tensorflow/python/keras/utils:data_utils",
],
)
py_library(
name = "sequence",
srcs = [
"sequence.py",
],
deps = [
"//tensorflow/python:util",
"//tensorflow/python/keras/utils:data_utils",
],
)
py_library(
name = "text",
srcs = [
"text.py",
],
deps = ["//tensorflow/python:util"],
)
tf_py_test(
name = "image_test",
size = "medium",
srcs = ["image_test.py"],
python_version = "PY3",
deps = [
":image",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "sequence_test",
size = "small",
srcs = ["sequence_test.py"],
python_version = "PY3",
deps = [
":sequence",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
],
)
tf_py_test(
name = "text_test",
size = "small",
srcs = ["text_test.py"],
python_version = "PY3",
deps = [
":text",
"//tensorflow/python:client_testlib",
"//third_party/py/numpy",
],
)

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

@ -36,16 +36,23 @@ py_library(
],
)
py_library(
name = "data_utils",
srcs = ["data_utils.py"],
srcs_version = "PY2AND3",
deps = [":generic_utils"],
)
py_library(
name = "engine_utils",
srcs = [
"conv_utils.py",
"data_utils.py",
"io_utils.py",
"losses_utils.py",
],
srcs_version = "PY2AND3",
deps = [
":data_utils",
"//tensorflow/python/keras:backend",
"//tensorflow/python/ops/losses:loss_reduction",
],