torch/ao/nn/quantizable/modules/__init__.py

from .activation import MultiheadAttention
from .rnn import LSTM
from .rnn import LSTMCell

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