9dd3461
1
2
3
4
5
6
7
8
9
10
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', ]
