diff options
Diffstat (limited to 'tensorflow/models/image/alexnet/BUILD')
-rw-r--r-- | tensorflow/models/image/alexnet/BUILD | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/tensorflow/models/image/alexnet/BUILD b/tensorflow/models/image/alexnet/BUILD new file mode 100644 index 0000000000..e1b9cd6965 --- /dev/null +++ b/tensorflow/models/image/alexnet/BUILD @@ -0,0 +1,28 @@ +# Description: +# Benchmark for AlexNet. + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +py_binary( + name = "alexnet_benchmark", + srcs = [ + "alexnet_benchmark.py", + ], + deps = [ + "//tensorflow:tensorflow_py", + ], +) + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) |