experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Reenable LSTM test which were disabled because of flaky.

Browse Source 
PiperOrigin-RevId: 257038325
This commit is contained in:
Scott Zhu 2019-07-08 12:59:28 -07:00 committed by TensorFlower Gardener
parent 4b59b23d56
commit e6dda7def8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/keras/layers/lstm_v2_test.py
View File

@ -351,7 +351,7 @@ class LSTMV2Test(keras_parameterized.TestCase):
self.assertAllClose(y_2, y_4, rtol=1e-5, atol=2e-5)
@parameterized.named_parameters(('v0', 0), ('v1', 1), ('v2', 2))
def DISABLED_test_implementation_mode_LSTM(self, implementation_mode):
def test_implementation_mode_LSTM(self, implementation_mode):
num_samples = 2
timesteps = 3
embedding_dim = 4
@ -552,7 +552,7 @@ class LSTMV2Test(keras_parameterized.TestCase):
self.assertAllClose(y_1, y_2)
self.assertAllClose(y_2, y_3)
def DISABLED_test_return_sequences_LSTM(self):
def test_return_sequences_LSTM(self):
num_samples = 2
timesteps = 3
embedding_dim = 4