| |
| r"""Quantized Reference Modules |
| |
| This module is in the process of migration to |
| `torch/ao/nn/quantized/reference`, and is kept here for |
| compatibility while the migration process is ongoing. |
| If you are adding a new entry/functionality, please, add it to the |
| appropriate file under the `torch/ao/nn/quantized/reference`, |
| while adding an import statement here. |
| """ |
|
|
| from torch.ao.nn.quantized.reference.modules.rnn import RNNCellBase |
| from torch.ao.nn.quantized.reference.modules.rnn import RNNCell |
| from torch.ao.nn.quantized.reference.modules.rnn import LSTMCell |
| from torch.ao.nn.quantized.reference.modules.rnn import GRUCell |
| from torch.ao.nn.quantized.reference.modules.rnn import RNNBase |
| from torch.ao.nn.quantized.reference.modules.rnn import LSTM |
|
|
