models
======

.. automodule:: models

torch.nn.RNN-style interface
----------------------------

.. autoclass:: models.AdaptiveRNN
    :members:
    :inherited-members: Module

.. autoclass:: models.AdaptiveGRU
    :members:
    :inherited-members: Module

torch.nn.RNNCell-style interface
--------------------------------

.. autoclass:: models.AdaptiveRNNCell
    :members:
    :inherited-members: Module

.. autoclass:: models.AdaptiveGRUCell
    :members:
    :inherited-members: Module


Add adaptive computation time to an RNNCell
-------------------------------------------

.. autoclass:: models.AdaptiveCellWrapper
    :members: