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

Fix more build dependency and tests in keras for sync to OSS.

Browse Source 
PiperOrigin-RevId: 336947422
Change-Id: Ib6c829551b45050ace5a94aa4450a8f36fea46e6
This commit is contained in:
Scott Zhu 2020-10-13 13:47:20 -07:00 committed by TensorFlower Gardener
parent 7b2f9fe0a2
commit 0f53c3fd79
4 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/engine/BUILD
View File

@ -731,7 +731,7 @@ tf_py_test(
tf_py_test( tf_py_test(
name = "deferred_sequential_test", name = "deferred_sequential_test",
size = "small", size = "medium",
srcs = ["deferred_sequential_test.py"], srcs = ["deferred_sequential_test.py"],
python_version = "PY3", python_version = "PY3",
tags = [ tags = [

8
tensorflow/python/keras/layers/preprocessing/BUILD
View File

@ -476,7 +476,9 @@ distribute_py_test(
srcs = ["discretization_distribution_test.py"], srcs = ["discretization_distribution_test.py"],
main = "discretization_distribution_test.py", main = "discretization_distribution_test.py",
python_version = "PY3", python_version = "PY3",
tags = ["multi_and_single_gpu"], tags = [
"multi_and_single_gpu",
],
deps = [ deps = [
":discretization", ":discretization",
":preprocessing_test_utils", ":preprocessing_test_utils",
@ -618,7 +620,9 @@ distribute_py_test(
srcs = ["normalization_distribution_test.py"], srcs = ["normalization_distribution_test.py"],
main = "normalization_distribution_test.py", main = "normalization_distribution_test.py",
python_version = "PY3", python_version = "PY3",
tags = ["no_oss"], tags = [
"no_oss",
],
deps = [ deps = [
":normalization", ":normalization",
":preprocessing_test_utils", ":preprocessing_test_utils",

2
tensorflow/python/keras/tests/BUILD
View File

@ -454,6 +454,7 @@ tf_py_test(
"//tensorflow/python/eager:backprop", "//tensorflow/python/eager:backprop",
"//tensorflow/python/eager:context", "//tensorflow/python/eager:context",
"//tensorflow/python/eager:def_function", "//tensorflow/python/eager:def_function",
"//tensorflow/python/keras",
"//tensorflow/python/keras:combinations", "//tensorflow/python/keras:combinations",
"//tensorflow/python/keras/engine", "//tensorflow/python/keras/engine",
"//tensorflow/python/keras/layers:core", "//tensorflow/python/keras/layers:core",
@ -488,6 +489,7 @@ tf_py_test(
"//tensorflow/python/eager:context", "//tensorflow/python/eager:context",
"//tensorflow/python/eager:def_function", "//tensorflow/python/eager:def_function",
"//tensorflow/python/eager:test", "//tensorflow/python/eager:test",
"//tensorflow/python/keras",
"//tensorflow/python/keras:combinations", "//tensorflow/python/keras:combinations",
"//tensorflow/python/keras/engine", "//tensorflow/python/keras/engine",
"//tensorflow/python/keras/layers:core", "//tensorflow/python/keras/layers:core",

2
tensorflow/python/keras/tests/tracking_util_test.py
View File

@ -90,7 +90,7 @@ class InterfaceTests(test.TestCase):
def testSaveWithOnlyKerasSession(self): def testSaveWithOnlyKerasSession(self):
with ops.Graph().as_default(): with ops.Graph().as_default(), self.cached_session():
inp = input_layer.Input([1]) inp = input_layer.Input([1])
dense = core.Dense(1)(inp) dense = core.Dense(1)(inp)
model = training.Model(inp, dense) model = training.Model(inp, dense)