aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/tutorials/mnist/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/examples/tutorials/mnist/BUILD')
-rw-r--r--tensorflow/examples/tutorials/mnist/BUILD115
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__"],
+)