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

[TF2XLA] [NFC] Test case for uninitialized tensor array due to dynamic size

Browse Source 
PiperOrigin-RevId: 348732190
Change-Id: I419f2211a70e95e96bd9d44bd66ee271321b006a
This commit is contained in:
George Karpenkov 2020-12-22 19:53:56 -08:00 committed by TensorFlower Gardener
parent d0a597a129
commit 907bde43cc
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tensorflow/python/eager/def_function_xla_jit_test.py
View File

@ -871,6 +871,24 @@ class DefFunctionTest(xla_test.XLATestCase):
f(constant_op.constant(1)) f(constant_op.constant(1))
@test_util.disable_mlir_bridge('TODO(b/155782411): MLIR bridge does not'
'support stack traces')
def testTensorArrayErrorMessage(self):
with ops.device('device:{}:0'.format(self.device)):
@def_function.function(jit_compile=True)
def f():
ta = tensor_array_ops.TensorArray(
dtype=dtypes.float32,
size=2,
dynamic_size=True,
element_shape=(None,))
return ta.concat() # EXPECTED_MESSAGE
with self.assertRaisesRegex(errors.InvalidArgumentError,
'EXPECTED_MESSAGE'):
f()
if __name__ == '__main__': if __name__ == '__main__':
ops.enable_eager_execution() ops.enable_eager_execution()