Lines Matching full:lstm

44     """Returns a ScriptModule that mimics a PyTorch native LSTM."""
81     """Returns a ScriptModule that mimics a PyTorch native LSTM."""
163         # XXX: This is true for our LSTM / NLP use case and helps simplify code
358         # Introduces a Dropout layer on the outputs of each LSTM layer except
364                 "dropout lstm adds dropout layers after all but last "
409     # Control: pytorch native LSTM
410     lstm = nn.LSTM(input_size, hidden_size, 1)
412     for lstm_param, custom_param in zip(lstm.all_weights[0], rnn.parameters()):
416     lstm_out, lstm_out_state = lstm(inp, lstm_state)
433     # Control: pytorch native LSTM
434     lstm = nn.LSTM(input_size, hidden_size, num_layers)
438         for lstm_param, custom_param in zip(lstm.all_weights[layer], custom_params):
442     lstm_out, lstm_out_state = lstm(inp, lstm_state)
462     # Control: pytorch native LSTM
463     lstm = nn.LSTM(input_size, hidden_size, num_layers, bidirectional=True)
469             for lstm_param, custom_param in zip(lstm.all_weights[index], custom_params):
473     lstm_out, lstm_out_state = lstm(inp, lstm_state)