Searched defs:grad_cy (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | RNN.cu | 407 void lstm_backward_impl(const Tensor& grad_hy, const Tensor& grad_cy, in lstm_backward_impl()
548 void checkLSTMBackwardSizes(const TensorArg& grad_hy, const TensorArg& grad_cy, in checkLSTMBackwardSizes()
573 const Tensor& grad_cy = c10::value_or_else(grad_cy_opt, [] {return Tensor();}); in _thnn_fused_lstm_cell_backward_impl_cuda() local
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | RNN_miopen.cpp | 560 const Tensor& grad_cy, in miopen_rnn_backward_input()
779 …auto grad_cy = cx.defined() ? (grad_cy_r.defined() ? grad_cy_r : at::zeros_like(cx, LEGACY_CONTIGU… in miopen_rnn_backward() local
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | RNN.cpp | 326 …auto grad_cy = cx_tmp.defined() ? (grad_cy_r.defined() ? grad_cy_r.contiguous() : at::zeros_like(c… in mkldnn_rnn_layer_backward() local
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | RNN.cpp | 1677 const Tensor& grad_cy, in _cudnn_rnn_backward_input()
2136 auto grad_cy = cx.defined() in _cudnn_rnn_backward() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 1552 const Tensor& grad_cy = c10::value_or_else(grad_cy_opt, [] {return Tensor();}); in _thnn_differentiable_lstm_cell_backward() local
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 6085 def checkLSTMBackwardSizes(grad_hy, grad_cy, cx, cy, workspace): argument
6101 def _thnn_fused_lstm_cell_backward_impl(grad_hy, grad_cy, cx, cy, workspace, has_bias): argument
|
| /external/pytorch/test/ |
| D | test_nn.py | 3966 cx_val=None, grad_cy=None): argument
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 7065 auto grad_cy = cx_tmp.defined() in mkldnn_rnn_layer_differentiable_backward() local
|