Searched defs:grad_hx (Results 1 – 2 of 2) sorted by relevance
480 … const Tensor& grad_input_gates, const Tensor& grad_hidden_gates, const Tensor& grad_hx) { in gru_backward_impl()646 auto grad_hx = at::empty_like(grad_hy, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in _thnn_fused_gru_cell_backward_cuda() local
1626 Tensor grad_hx = grad_hy * ig; in _thnn_differentiable_gru_cell_backward() local