BUILD 504 B

123456789101112131415161718192021222324252627282930
  1. # Description:
  2. # Benchmark for AlexNet.
  3. licenses(["notice"]) # Apache 2.0
  4. exports_files(["LICENSE"])
  5. py_binary(
  6. name = "alexnet_benchmark",
  7. srcs = [
  8. "alexnet_benchmark.py",
  9. ],
  10. srcs_version = "PY2AND3",
  11. deps = [
  12. "//tensorflow:tensorflow_py",
  13. ],
  14. )
  15. filegroup(
  16. name = "all_files",
  17. srcs = glob(
  18. ["**/*"],
  19. exclude = [
  20. "**/METADATA",
  21. "**/OWNERS",
  22. ],
  23. ),
  24. visibility = ["//tensorflow:__subpackages__"],
  25. )