Searched defs:hidden_gates (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| 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()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 1546 …const Tensor& hidden_gates, const std::optional<Tensor>& input_bias_opt, const std::optional<Tenso… in _thnn_differentiable_lstm_cell_backward()
1601 const Tensor& hidden_gates, in _thnn_differentiable_gru_cell_backward()
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 5864 hidden_gates, argument
5904 hidden_gates, argument
|