Searched defs:grad_cx (Results 1 – 1 of 1) sorted by relevance
410 const Tensor& grad_gates, const Tensor& grad_cx) { in lstm_backward_impl()583 auto grad_cx = at::empty_like(cx, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in _thnn_fused_lstm_cell_backward_impl_cuda() local