Disable TF32 execution for lstm_test.
diff --git a/tensorflow/compiler/tests/lstm_test.py b/tensorflow/compiler/tests/lstm_test.py
index 77d5fcb..1096b8c 100644
--- a/tensorflow/compiler/tests/lstm_test.py
+++ b/tensorflow/compiler/tests/lstm_test.py
@@ -25,6 +25,7 @@
from tensorflow.python.framework import constant_op
from tensorflow.python.framework import dtypes
from tensorflow.python.framework import ops
+from tensorflow.python.framework import test_util
from tensorflow.python.ops import array_ops
from tensorflow.python.ops import gradients_impl
from tensorflow.python.ops import init_ops
@@ -87,6 +88,8 @@
self.evaluate(variables.global_variables_initializer())
return self.evaluate([m, c])
+ @test_util.run_without_tensor_float_32("TF32 capable devices fail the test"
+ " due to reduced matmul precision")
def testLSTMCell(self):
# Run with all-0 weights, no padding.
m, c = self._RunLSTMCell('zeros', init_ops.zeros_initializer(), 0., 0., 0.)
@@ -172,6 +175,8 @@
self.evaluate(variables.global_variables_initializer())
return self.evaluate(out_seq)
+ @test_util.run_without_tensor_float_32("TF32 capable devices fail the test"
+ " due to reduced matmul precision")
def testLSTMLayer(self):
# Run with all-0 weights, no padding.
o = self._RunLSTMLayer('zeros', init_ops.zeros_initializer(), 0., 0., 0.)
