Searched refs:lstm_forward_impl (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | RNN.cu | 370 void lstm_forward_impl(const Tensor& input_gates, const Tensor& hidden_gates, in lstm_forward_impl() function
540 …lstm_forward_impl<scalar_t, int32_t>(input_gates, hidden_gates, input_bias, hidden_bias, cx, hy, c… in _thnn_fused_lstm_cell_cuda()
542 …lstm_forward_impl<scalar_t, int64_t>(input_gates, hidden_gates, input_bias, hidden_bias, cx, hy, c… in _thnn_fused_lstm_cell_cuda()
|