aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/models/image/cifar10/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/models/image/cifar10/BUILD')
-rw-r--r--tensorflow/models/image/cifar10/BUILD79
1 files changed, 79 insertions, 0 deletions
diff --git a/tensorflow/models/image/cifar10/BUILD b/tensorflow/models/image/cifar10/BUILD
new file mode 100644
index 0000000000..adf9aaffd4
--- /dev/null
+++ b/tensorflow/models/image/cifar10/BUILD
@@ -0,0 +1,79 @@
+# Description:
+# Example TensorFlow models for CIFAR-10
+
+licenses(["notice"]) # Apache 2.0
+
+exports_files(["LICENSE"])
+
+py_library(
+ name = "cifar10_input",
+ srcs = ["cifar10_input.py"],
+ deps = [
+ "//tensorflow:tensorflow_py",
+ ],
+)
+
+py_test(
+ name = "cifar10_input_test",
+ srcs = ["cifar10_input_test.py"],
+ deps = [
+ ":cifar10_input",
+ "//tensorflow:tensorflow_py",
+ "//tensorflow/python:framework_test_lib",
+ "//tensorflow/python:platform_test",
+ ],
+)
+
+py_library(
+ name = "cifar10",
+ srcs = ["cifar10.py"],
+ deps = [
+ ":cifar10_input",
+ "//tensorflow:tensorflow_py",
+ ],
+)
+
+py_binary(
+ name = "cifar10_eval",
+ srcs = [
+ "cifar10_eval.py",
+ ],
+ visibility = ["//tensorflow:__subpackages__"],
+ deps = [
+ ":cifar10",
+ ],
+)
+
+py_binary(
+ name = "cifar10_train",
+ srcs = [
+ "cifar10_train.py",
+ ],
+ visibility = ["//tensorflow:__subpackages__"],
+ deps = [
+ ":cifar10",
+ ],
+)
+
+py_binary(
+ name = "cifar10_multi_gpu_train",
+ srcs = [
+ "cifar10_multi_gpu_train.py",
+ ],
+ visibility = ["//tensorflow:__subpackages__"],
+ deps = [
+ ":cifar10",
+ ],
+)
+
+filegroup(
+ name = "all_files",
+ srcs = glob(
+ ["**/*"],
+ exclude = [
+ "**/METADATA",
+ "**/OWNERS",
+ ],
+ ),
+ visibility = ["//tensorflow:__subpackages__"],
+)