Searched defs:_cudnn_rnn_backward (Results 1 – 2 of 2) sorted by relevance
21 def _cudnn_rnn_backward(op, *grads): function
70 std::tuple<Tensor, Tensor, Tensor, std::vector<Tensor>> _cudnn_rnn_backward( in _cudnn_rnn_backward() function2088 std::tuple<Tensor, Tensor, Tensor, std::vector<Tensor>> _cudnn_rnn_backward( in _cudnn_rnn_backward() function