Searched refs:dprev_h (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 7110 Tensor dx, dWx, dWh, db, db_, dprev_h, dprev_c, dWh_, dWx_; in mkldnn_rnn_layer_differentiable_backward() local
7138 dprev_h = at::matmul(da, weight1); in mkldnn_rnn_layer_differentiable_backward()
7151 grad_hy = dprev_h; in mkldnn_rnn_layer_differentiable_backward()
7156 return std::make_tuple(cat_layer_dx, dWx, dWh, db, db, dprev_h, dprev_c); in mkldnn_rnn_layer_differentiable_backward()
|