(i, num_elems, *args)
| 1165 | |
| 1166 | # TODO(tombagby): Update to tfe.defun |
| 1167 | def cond(i, num_elems, *args): |
| 1168 | del args |
| 1169 | return i >= 0 if reverse else i < num_elems |
| 1170 | |
| 1171 | # The loop *args are [output tensors] + [accumulator tensors] which must |
| 1172 | # be paired. Each output corresponds to one accumulator. |
no outgoing calls