Home
last modified time | relevance | path

Searched defs:grad_output_r (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp559 const Tensor& output_r, const Tensor& grad_output_r, const Tensor& grad_hy, in miopen_rnn_backward_input()
769 const Tensor& grad_output_r = c10::value_or_else(grad_output_r_opt, [] {return Tensor();}); in miopen_rnn_backward() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp1675 const Tensor& grad_output_r, in _cudnn_rnn_backward_input()
2115 const Tensor& grad_output_r = in _cudnn_rnn_backward() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DRNN.cpp318 const Tensor& grad_output_r = c10::value_or_else(grad_output_r_opt, [] {return Tensor();}); in mkldnn_rnn_layer_backward() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp7048 const Tensor& grad_output_r = in mkldnn_rnn_layer_differentiable_backward() local