Searched defs:grad_output_r_opt (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | RNN.cpp | 58 const std::optional<Tensor>& grad_output_r_opt, in mkldnn_rnn_layer_backward()
305 const std::optional<Tensor>& grad_output_r_opt, in mkldnn_rnn_layer_backward()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 42 …const Tensor& output, const std::optional<Tensor>& grad_output_r_opt, const std::optional<Tensor>&… in miopen_rnn_backward()
762 …const Tensor& output, const std::optional<Tensor>& grad_output_r_opt, const std::optional<Tensor>&… in miopen_rnn_backward()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 78 const std::optional<Tensor>& grad_output_r_opt, in _cudnn_rnn_backward()
2096 const std::optional<Tensor>& grad_output_r_opt, in _cudnn_rnn_backward()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 6169 grad_output_r_opt, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 7035 const std::optional<Tensor>& grad_output_r_opt, in mkldnn_rnn_layer_differentiable_backward()
|