Searched defs:grad_input_gates (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()644 auto grad_input_gates = at::empty({workspace.size(0), hidden_size * 3}, workspace.options()); in _thnn_fused_gru_cell_backward_cuda() local
1632 Tensor grad_input_gates = at::cat({grg,gig,std::move(gin)}, 1); in _thnn_differentiable_gru_cell_backward() local