diff options
Diffstat (limited to 'tensorflow/contrib/seq2seq/BUILD')
-rw-r--r-- | tensorflow/contrib/seq2seq/BUILD | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/contrib/seq2seq/BUILD b/tensorflow/contrib/seq2seq/BUILD index 9566d03211..3c314e2f28 100644 --- a/tensorflow/contrib/seq2seq/BUILD +++ b/tensorflow/contrib/seq2seq/BUILD @@ -41,6 +41,18 @@ cuda_py_test( ) cuda_py_test( + name = "loss_test", + size = "medium", + srcs = ["python/kernel_tests/loss_test.py"], + additional_deps = [ + ":seq2seq_py", + "//tensorflow:tensorflow_py", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:platform_test", + ], +) + +cuda_py_test( name = "seq2seq_test", size = "medium", srcs = ["python/kernel_tests/seq2seq_test.py"], |