diff options
| author |  Scott Zhu <scottzhu@google.com> | 2018-08-15 09:49:41 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-15 09:55:00 -0700 |
| commit | 0c71f791bf353e84e9a46487c81914d250470210 (patch) | |
| tree | 26dbf5e41087ab913d17bde06022affd3be82387 /tensorflow/contrib/rnn | |
| parent | fabf7a5c18555e52cef75643b4d4532ff70aa23e (diff) | |
Split unit test case for further parallelize.
PiperOrigin-RevId: 208834783
Diffstat (limited to 'tensorflow/contrib/rnn')
| -rw-r--r-- | tensorflow/contrib/rnn/BUILD | 2 | ||||
| -rw-r--r-- | tensorflow/contrib/rnn/python/kernel_tests/core_rnn_test.py | 5 |
2 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/contrib/rnn/BUILD b/tensorflow/contrib/rnn/BUILD index 1a1be5dc70..5874245d58 100644 --- a/tensorflow/contrib/rnn/BUILD +++ b/tensorflow/contrib/rnn/BUILD @@ -149,7 +149,7 @@ cuda_py_tests( cuda_py_tests( name = "core_rnn_test", - size = "large", + size = "medium", srcs = ["python/kernel_tests/core_rnn_test.py"], additional_deps = [ ":rnn_py", diff --git a/tensorflow/contrib/rnn/python/kernel_tests/core_rnn_test.py b/tensorflow/contrib/rnn/python/kernel_tests/core_rnn_test.py index f7088c6331..d62ec45d18 100644 --- a/tensorflow/contrib/rnn/python/kernel_tests/core_rnn_test.py +++ b/tensorflow/contrib/rnn/python/kernel_tests/core_rnn_test.py @@ -1287,9 +1287,12 @@ class LSTMTest(test.TestCase): @test_util.run_in_graph_and_eager_modes def testDynamicEquivalentToStaticRNN(self): - self._testDynamicEquivalentToStaticRNN(use_sequence_length=True) self._testDynamicEquivalentToStaticRNN(use_sequence_length=False) + @test_util.run_in_graph_and_eager_modes + def testDynamicEquivalentToStaticRNNWithSequenceLength(self): + self._testDynamicEquivalentToStaticRNN(use_sequence_length=True) + class BidirectionalRNNTest(test.TestCase): |