Searched defs:grad_hy (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()
479 void gru_backward_impl(const Tensor& grad_hy, const Tensor& workspace, in gru_backward_impl()
548 void checkLSTMBackwardSizes(const TensorArg& grad_hy, const TensorArg& grad_cy, in checkLSTMBackwardSizes()
572 const Tensor& grad_hy = *grad_hy_maybe_owned; in _thnn_fused_lstm_cell_backward_impl_cuda() local
633 void checkGRUBackwardSizes(const TensorArg& grad_hy, const TensorArg& workspace) { in checkGRUBackwardSizes()
640 const Tensor& grad_hy, const Tensor& workspace, bool has_bias) { in _thnn_fused_gru_cell_backward_cuda()
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | RNN_miopen.cpp | 559 const Tensor& output_r, const Tensor& grad_output_r, const Tensor& grad_hy, in miopen_rnn_backward_input()
778 …auto grad_hy = grad_hy_r.defined() ? grad_hy_r : at::zeros_like(hx, LEGACY_CONTIGUOUS_MEMORY_FORMA… in miopen_rnn_backward() local
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | RNN.cpp | 325 …auto grad_hy = grad_hy_r.defined() ? grad_hy_r.contiguous() : at::zeros_like(hx_, LEGACY_CONTIGUOU… in mkldnn_rnn_layer_backward() local
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | RNN.cpp | 1676 const Tensor& grad_hy, in _cudnn_rnn_backward_input()
2133 auto grad_hy = grad_hy_r.defined() in _cudnn_rnn_backward() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 1551 const Tensor& grad_hy = *grad_hy_maybe_owned; in _thnn_differentiable_lstm_cell_backward() local
1599 const Tensor& grad_hy, in _thnn_differentiable_gru_cell_backward()
|
| /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 | 3965 def forward_backward(cuda, rnn, input_val, grad_output, weights_val, hx_val, grad_hy, argument
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 7062 auto grad_hy = grad_hy_r.defined() in mkldnn_rnn_layer_differentiable_backward() local
|