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:
parent
af04b99a42
commit
c10283ae80
@ -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))
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user