|
|
|
|
|
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 |
|
|
|
