(*args)
| 336 | |
| 337 | @function.Defun(*_Dtypes(fwd_sig)) |
| 338 | def Fwd(*args): |
| 339 | (theta, state0, inputs) = _Pack(args, fwd_sig) |
| 340 | state1, extras = self._cell_fn(theta, state0, inputs) |
| 341 | assert not function.get_extra_args(), ( |
| 342 | 'cell_fn is not pure with extra args: %s.' % |
| 343 | (function.get_extra_args())) |
| 344 | _AssertIsCompatible(state1, self._state) |
| 345 | _AssertIsCompatible(extras, self._extras) |
| 346 | return _Flatten([state1, extras]) |
| 347 | |
| 348 | # Wraps cell_fn in a TF Function as a for-loop's body. |
| 349 | # |
nothing calls this directly
no test coverage detected