Searched refs:script_lstm (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | custom_lstms.py | 35 def script_lstm( function
429 rnn = script_lstm(input_size, hidden_size, num_layers)
458 rnn = script_lstm(input_size, hidden_size, num_layers, bidirectional=True)
488 rnn = script_lstm(input_size, hidden_size, num_layers, dropout=True)
|
| H A D | factory.py | 112 from .custom_lstms import LSTMState, script_lstm
119 ge = script_lstm(input_size, hidden_size, num_layers, dropout=True).cuda()
|