Searched refs:_thnn_fused_lstm_cell_backward_impl (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/expect/ |
H A D | HasDecompTest.test_has_decomposition.expect | 587 aten::_thnn_fused_lstm_cell_backward_impl 588 aten::_thnn_fused_lstm_cell_backward_impl.out
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | RNN.cpp | 1176 …auto ret = at::_thnn_fused_lstm_cell_backward_impl(grad_hy_opt, grad_cy_opt, cx, cy, workspace, ha… in _thnn_fused_lstm_cell_backward()
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _meta_registrations.py | 6100 @register_meta(aten._thnn_fused_lstm_cell_backward_impl.default) 6101 def _thnn_fused_lstm_cell_backward_impl(grad_hy, grad_cy, cx, cy, workspace, has_bias): function
|