Clean up and expand integration tests to cover shape, dtype and structure verification across different loop types.

PiperOrigin-RevId: 290602557
Change-Id: I2e2d97db4cdca49f4ff5c29e9083fb55c9fe6a9b
This commit is contained in:
Dan Moldovan 2020-01-20 05:38:55 -08:00 committed by TensorFlower Gardener
parent af04b99a42
commit c10283ae80

View File

@ -188,7 +188,7 @@ def _verify_single_loop_var(
shape_invariant))
if not _is_subshape(exit_shape, shape_invariant):
raise ValueError(
'"{}" has shape {} after the loop, which does not conform with'
'"{}" has shape {} after one iteration, which does not conform with'
' the shape invariant {}.'.format(
name, exit_shape, shape_invariant))