diff options
Diffstat (limited to 'tensorflow/models/image/cifar10/BUILD')
-rw-r--r-- | tensorflow/models/image/cifar10/BUILD | 79 |
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__"], +) |