diff options
Diffstat (limited to 'tensorflow/models/rnn/translate/BUILD')
-rw-r--r-- | tensorflow/models/rnn/translate/BUILD | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/tensorflow/models/rnn/translate/BUILD b/tensorflow/models/rnn/translate/BUILD new file mode 100644 index 0000000000..0899bf689e --- /dev/null +++ b/tensorflow/models/rnn/translate/BUILD @@ -0,0 +1,71 @@ +# Description: +# Example neural translation models. + +package(default_visibility = ["//visibility:public"]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +py_library( + name = "data_utils", + srcs = [ + "data_utils.py", + ], + deps = [ + "//tensorflow:tensorflow_py", + ], +) + +py_library( + name = "seq2seq_model", + srcs = [ + "seq2seq_model.py", + ], + deps = [ + ":data_utils", + "//tensorflow:tensorflow_py", + "//tensorflow/models/rnn:seq2seq", + ], +) + +py_binary( + name = "translate", + srcs = [ + "translate.py", + ], + deps = [ + ":data_utils", + ":seq2seq_model", + "//tensorflow:tensorflow_py", + ], +) + +py_test( + name = "translate_test", + size = "medium", + srcs = [ + "translate.py", + ], + args = [ + "--self_test=True", + ], + main = "translate.py", + deps = [ + ":data_utils", + ":seq2seq_model", + "//tensorflow:tensorflow_py", + ], +) + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) |