1from torch.ao.nn.quantizable.modules.activation import MultiheadAttention 2from torch.ao.nn.quantizable.modules.rnn import LSTM, LSTMCell 3 4 5__all__ = [ 6 "LSTM", 7 "LSTMCell", 8 "MultiheadAttention", 9] 10