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

Update tests to use tf_py_test.

Browse Source 
PiperOrigin-RevId: 261234665
This commit is contained in:
A. Unique TensorFlower 2019-08-01 17:48:36 -07:00 committed by TensorFlower Gardener
parent ef9f0e8f2f
commit 2859264874
1 changed files with 12 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
tensorflow/examples/tutorials/mnist/BUILD
View File

@ -1,7 +1,7 @@
# Description: # Description:
# Example TensorFlow models for MNIST used in tutorials # Example TensorFlow models for MNIST used in tutorials
load("//tensorflow:tensorflow.bzl", "py_test") load("//tensorflow:tensorflow.bzl", "tf_py_test")
package( package(
licenses = ["notice"], # Apache 2.0 licenses = ["notice"], # Apache 2.0
@ -94,43 +94,38 @@ py_binary(
], ],
) )
py_test( tf_py_test(
name = "fully_connected_feed_test", name = "fully_connected_feed_test",
size = "medium",
srcs = [ srcs = [
"fully_connected_feed.py", "fully_connected_feed.py",
], ],
additional_deps = [
":input_data",
":mnist",
"//tensorflow:tensorflow_py",
],
args = [ args = [
"--fake_data", "--fake_data",
"--max_steps=10", "--max_steps=10",
], ],
main = "fully_connected_feed.py", main = "fully_connected_feed.py",
python_version = "PY2",
srcs_version = "PY2AND3",
deps = [
":input_data",
":mnist",
"//tensorflow:tensorflow_py",
],
) )
py_test( tf_py_test(
name = "mnist_with_summaries_test", name = "mnist_with_summaries_test",
size = "small", size = "small",
srcs = [ srcs = [
"mnist_with_summaries.py", "mnist_with_summaries.py",
], ],
additional_deps = [
":input_data",
"//tensorflow:tensorflow_py",
],
args = [ args = [
"--fake_data", "--fake_data",
"--max_steps=10", "--max_steps=10",
"--learning_rate=0.00", "--learning_rate=0.00",
], ],
main = "mnist_with_summaries.py", main = "mnist_with_summaries.py",
python_version = "PY2",
srcs_version = "PY2AND3",
tags = ["notsan"], # http://b/29184009 tags = ["notsan"], # http://b/29184009
deps = [
":input_data",
"//tensorflow:tensorflow_py",
],
) )