diff options
Diffstat (limited to 'tensorflow/examples/tutorials/mnist/BUILD')
-rw-r--r-- | tensorflow/examples/tutorials/mnist/BUILD | 115 |
1 files changed, 115 insertions, 0 deletions
diff --git a/tensorflow/examples/tutorials/mnist/BUILD b/tensorflow/examples/tutorials/mnist/BUILD new file mode 100644 index 0000000000..4fc90730d5 --- /dev/null +++ b/tensorflow/examples/tutorials/mnist/BUILD @@ -0,0 +1,115 @@ +# Description: +# Example TensorFlow models for MNIST used in tutorials + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +py_library( + name = "package", + srcs = [ + "__init__.py", + ], + srcs_version = "PY2AND3", + visibility = ["//tensorflow:__subpackages__"], + deps = [ + ":input_data", + ":mnist", + ], +) + +py_library( + name = "input_data", + srcs = ["input_data.py"], + srcs_version = "PY2AND3", + visibility = ["//tensorflow:__subpackages__"], + deps = ["//tensorflow:tensorflow_py"], +) + +py_library( + name = "mnist", + srcs = [ + "mnist.py", + ], + srcs_version = "PY2AND3", + visibility = ["//tensorflow:__subpackages__"], + deps = [ + "//tensorflow:tensorflow_py", + ], +) + +py_binary( + name = "fully_connected_feed", + srcs = [ + "fully_connected_feed.py", + ], + srcs_version = "PY2AND3", + deps = [ + ":input_data", + ":mnist", + "//tensorflow:tensorflow_py", + ], +) + +py_binary( + name = "mnist_with_summaries", + srcs = [ + "mnist_with_summaries.py", + ], + srcs_version = "PY2AND3", + deps = [ + ":input_data", + "//tensorflow:tensorflow_py", + ], +) + +py_test( + name = "fully_connected_feed_test", + size = "small", + srcs = [ + "fully_connected_feed.py", + ], + args = [ + "--fake_data", + "--max_steps=10", + "--train_dir=/tmp/mnist", + ], + main = "fully_connected_feed.py", + srcs_version = "PY2AND3", + deps = [ + ":input_data", + ":mnist", + "//tensorflow:tensorflow_py", + ], +) + +py_test( + name = "mnist_with_summaries_test", + size = "small", + srcs = [ + "mnist_with_summaries.py", + ], + args = [ + "--fake_data", + "--max_steps=10", + "--learning_rate=0.00", + ], + main = "mnist_with_summaries.py", + srcs_version = "PY2AND3", + deps = [ + ":input_data", + "//tensorflow:tensorflow_py", + ], +) + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) |