(*aug_loop_vars)
| 751 | return test(*aug_loop_vars[loop_vars_slice]) |
| 752 | |
| 753 | def aug_body(*aug_loop_vars): |
| 754 | state = aug_loop_vars[state_slice] |
| 755 | set_state(state) |
| 756 | loop_vars = body(*aug_loop_vars[loop_vars_slice]) |
| 757 | new_state = loop_vars + get_state() |
| 758 | _verify_tf_loop_vars(aug_loop_vars, new_state, basic_symbol_names, |
| 759 | composite_symbol_names) |
| 760 | |
| 761 | return new_state |
| 762 | |
| 763 | # Non-v2 while_loop unpacks the results when there is only one return value. |
| 764 | # This enforces consistency across versions. |
nothing calls this directly
no test coverage detected