diff options
Diffstat (limited to 'tensorflow/contrib/seq2seq/BUILD')
-rw-r--r-- | tensorflow/contrib/seq2seq/BUILD | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/seq2seq/BUILD b/tensorflow/contrib/seq2seq/BUILD index d5f0abe6a7..178bd81afb 100644 --- a/tensorflow/contrib/seq2seq/BUILD +++ b/tensorflow/contrib/seq2seq/BUILD @@ -16,9 +16,9 @@ py_library( ) cuda_py_test( - name = "layers_test", + name = "decoder_fn_test", size = "medium", - srcs = ["python/kernel_tests/layers_test.py"], + srcs = ["python/kernel_tests/decoder_fn_test.py"], additional_deps = [ ":seq2seq_py", "//tensorflow:tensorflow_py", @@ -28,9 +28,9 @@ cuda_py_test( ) cuda_py_test( - name = "loss_test", + name = "seq2seq_test", size = "medium", - srcs = ["python/kernel_tests/loss_test.py"], + srcs = ["python/kernel_tests/seq2seq_test.py"], additional_deps = [ ":seq2seq_py", "//tensorflow:tensorflow_py", |