1 # Description: 2 # Contains library to create sequence-to-sequence models on top of TensorFlow. 3 # APIs here are meant to evolve over time. 4 5 licenses(["notice"]) # Apache 2.0 6 7 exports_files(["LICENSE"]) 8 9 package(default_visibility = ["//visibility:public"]) 10 11 load("//tensorflow:tensorflow.bzl", "cuda_py_tests") 12 13 py_library( 14 name = "seq2seq_py", 15 srcs = [ 16 "__init__.py", 17 "python/__init__.py", 18 ] + glob( 19 ["python/ops/*.py"], 20 exclude = ["python/ops/**/*_test.py"], 21 ), 22 srcs_version = "PY2AND3", 23 visibility = ["//visibility:public"], 24 deps = [ 25 "//tensorflow/contrib/rnn:rnn_py", 26 "//tensorflow/python:array_ops", 27 "//tensorflow/python:control_flow_ops", 28 "//tensorflow/python:embedding_ops", 29 "//tensorflow/python:framework_for_generated_wrappers", 30 "//tensorflow/python:math_ops", 31 "//tensorflow/python:nn_ops", 32 "//tensorflow/python:rnn", 33 "//tensorflow/python:rnn_cell", 34 "//tensorflow/python:util", 35 "//tensorflow/python:variable_scope", 36 ], 37 ) 38 39 cuda_py_tests( 40 name = "seq2seq_test", 41 size = "medium", 42 srcs = ["python/kernel_tests/seq2seq_test.py"], 43 additional_deps = [ 44 ":seq2seq_py", 45 "//third_party/py/numpy", 46 "//tensorflow/contrib/rnn:rnn_py", 47 "//tensorflow/python:array_ops", 48 "//tensorflow/python:client_testlib", 49 "//tensorflow/python:clip_ops", 50 "//tensorflow/python:framework", 51 "//tensorflow/python:framework_for_generated_wrappers", 52 "//tensorflow/python:gradients", 53 "//tensorflow/python:init_ops", 54 "//tensorflow/python:nn", 55 "//tensorflow/python:rnn", 56 "//tensorflow/python:state_ops", 57 "//tensorflow/python:training", 58 "//tensorflow/python:variable_scope", 59 "//tensorflow/python:variables", 60 ], 61 tags = ["noasan"], # times out b/63678675 62 ) 63 64 filegroup( 65 name = "all_files", 66 srcs = glob( 67 ["**/*"], 68 exclude = [ 69 "**/METADATA", 70 "**/OWNERS", 71 ], 72 ), 73 visibility = ["//tensorflow:__subpackages__"], 74 ) 75