BUILD 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. # Description:
  2. # Example RNN models, including language models and sequence-to-sequence models.
  3. package(default_visibility = ["//visibility:public"])
  4. licenses(["notice"]) # Apache 2.0
  5. exports_files(["LICENSE"])
  6. py_library(
  7. name = "linear",
  8. srcs = [
  9. "linear.py",
  10. ],
  11. srcs_version = "PY2AND3",
  12. deps = [
  13. "//tensorflow:tensorflow_py",
  14. ],
  15. )
  16. py_library(
  17. name = "rnn_cell",
  18. srcs = [
  19. "rnn_cell.py",
  20. ],
  21. srcs_version = "PY2AND3",
  22. deps = [
  23. ":linear",
  24. "//tensorflow:tensorflow_py",
  25. ],
  26. )
  27. py_library(
  28. name = "package",
  29. srcs = [
  30. "__init__.py",
  31. ],
  32. srcs_version = "PY2AND3",
  33. deps = [
  34. ":rnn",
  35. ":rnn_cell",
  36. ":seq2seq",
  37. ],
  38. )
  39. py_library(
  40. name = "rnn",
  41. srcs = [
  42. "rnn.py",
  43. ],
  44. srcs_version = "PY2AND3",
  45. deps = [
  46. ":rnn_cell",
  47. "//tensorflow:tensorflow_py",
  48. ],
  49. )
  50. py_library(
  51. name = "seq2seq",
  52. srcs = [
  53. "seq2seq.py",
  54. ],
  55. srcs_version = "PY2AND3",
  56. deps = [
  57. ":rnn",
  58. "//tensorflow:tensorflow_py",
  59. ],
  60. )
  61. filegroup(
  62. name = "all_files",
  63. srcs = glob(
  64. ["**/*"],
  65. exclude = [
  66. "**/METADATA",
  67. "**/OWNERS",
  68. ],
  69. ),
  70. visibility = ["//tensorflow:__subpackages__"],
  71. )