Home
last modified time | relevance | path

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

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