Move test from while_v2_test.py to control_flow_ops_py_test.py.
This is to make sure we have coverage of explicitly instantiating a TensorArray and passing it to a while_loop in both v1 and v2 control flow, in a way they're actually called (while_v2_test.py calls while_v2 directly, instead of going through control_flow_ops.while_loop). PiperOrigin-RevId: 225913951
Loading
Please sign in to comment