提交 00786143 authored 作者: Boris Fomitchev's avatar Boris Fomitchev 提交者: notoraptor

Squashed commit of NVIDIA changes for CUDA9/Volta

上级 6bbe6a92
...@@ -30,7 +30,7 @@ int dnn_rnn_desc(int hidden_size, int num_layers, ...@@ -30,7 +30,7 @@ int dnn_rnn_desc(int hidden_size, int num_layers,
return -1; return -1;
} }
err = cudnnSetRNNDescriptor(handle, desc, hidden_size, num_layers, ddesc, err = cudnnSetRNNDescriptor(_handle, desc, hidden_size, num_layers, ddesc,
(cudnnRNNInputMode_t)input_mode, (cudnnRNNInputMode_t)input_mode,
(cudnnDirectionMode_t)direction_mode, (cudnnDirectionMode_t)direction_mode,
(cudnnRNNMode_t)rnn_mode, CUDNN_RNN_ALGO_STANDARD, data_type); (cudnnRNNMode_t)rnn_mode, CUDNN_RNN_ALGO_STANDARD, data_type);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论