Home
last modified time | relevance | path

Searched defs:grad_hy_opt (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DRNN.cu567 …nsor, Tensor> _thnn_fused_lstm_cell_backward_impl_cuda( const std::optional<Tensor>& grad_hy_opt, … in _thnn_fused_lstm_cell_backward_impl_cuda()
/external/pytorch/aten/src/ATen/native/
DRNN.cpp1172 …Tensor, Tensor, Tensor> _thnn_fused_lstm_cell_backward( const std::optional<Tensor>& grad_hy_opt, … in _thnn_fused_lstm_cell_backward()
1544 _thnn_differentiable_lstm_cell_backward( const std::optional<Tensor>& grad_hy_opt, const std::optio… in _thnn_differentiable_lstm_cell_backward()