(scan_state)
| 151 | num_steps -= 1 |
| 152 | |
| 153 | def cond_fn(scan_state) -> jnp.bool_: |
| 154 | _, _, i, _ = scan_state |
| 155 | return i < num_steps |
| 156 | |
| 157 | def loop_fn(scan_state: Any) -> Tuple[Any, Any, Any, Any]: |
| 158 | (dstate, input_token, i, _) = scan_state |
nothing calls this directly
no outgoing calls
no test coverage detected