torch/nn/quantizable/modules/__init__.py

from torch.ao.nn.quantizable.modules.activation import MultiheadAttention
from torch.ao.nn.quantizable.modules.rnn import LSTM
from torch.ao.nn.quantizable.modules.rnn import LSTMCell

__all__ = [
    'LSTM',
    'LSTMCell',
    'MultiheadAttention',
]