Searched refs:lstm_backward_setup (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | factory.py | 41 def lstm_backward_setup(lstm_outputs, seed=None): function
64 backward_setup=lstm_backward_setup,
76 backward_setup=lstm_backward_setup,
105 backward_setup=lstm_backward_setup,
133 backward_setup=lstm_backward_setup,
145 backward_setup=lstm_backward_setup,
157 backward_setup=lstm_backward_setup,
169 backward_setup=lstm_backward_setup,
181 backward_setup=lstm_backward_setup,
261 backward_setup=lstm_backward_setup,
[all …]
|