Searched defs:input_gates (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | RNN.cu | 33 const TensorArg& input_gates, const TensorArg& hidden_gates, in checkSizes()
370 void lstm_forward_impl(const Tensor& input_gates, const Tensor& hidden_gates, in lstm_forward_impl()
443 void gru_forward_impl(const Tensor& input_gates, const Tensor& hidden_gates, in gru_forward_impl()
518 const Tensor& input_gates, const Tensor& hidden_gates, in _thnn_fused_lstm_cell_cuda()
604 const Tensor& input_gates, const Tensor& hidden_gates, in _thnn_fused_gru_cell_cuda()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 1545 const Tensor& input_gates, in _thnn_differentiable_lstm_cell_backward()
1600 const Tensor& input_gates, in _thnn_differentiable_gru_cell_backward()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 5863 input_gates, argument
5903 input_gates, argument
|