Searched defs:grad_gates (Results 1 – 2 of 2) sorted by relevance
410 const Tensor& grad_gates, const Tensor& grad_cx) { in lstm_backward_impl()582 auto grad_gates = at::empty_like(workspace, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in _thnn_fused_lstm_cell_backward_impl_cuda() local
1593 Tensor grad_gates = at::cat({std::move(gig), std::move(gfg), std::move(gcg), std::move(gog)}, 1); in _thnn_differentiable_lstm_cell_backward() local