Support should be added for `torch.nn.LSTM` and `torch.nn.LSTMCell` modules.
Support should be added for
torch.nn.LSTMandtorch.nn.LSTMCellmodules.