A wrapper that handles loop-carried captured inputs.
(*args)
| 649 | |
| 650 | @function.Defun(*cond_dtypes, func_name="%s_Wrapper" % cond.name) |
| 651 | def CondWrapper(*args): |
| 652 | """A wrapper that handles loop-carried captured inputs.""" |
| 653 | return cond(*args[:len(body.declared_input_types)]) |
| 654 | |
| 655 | ret = gen_functional_ops._while( |
| 656 | input_ + body.captured_inputs, |