singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [singa] nudles commented on a change in pull request #662: CUDNN LSTM
Date Sat, 11 Apr 2020 13:38:32 GMT
nudles commented on a change in pull request #662: CUDNN LSTM
URL: https://github.com/apache/singa/pull/662#discussion_r407064103
 
 

 ##########
 File path: src/api/model_operation.i
 ##########
 @@ -188,6 +189,50 @@ Tensor GpuPoolingForward(const CudnnPoolingHandle &cph, const Tensor
&x);
 
 Tensor GpuPoolingBackward(const CudnnPoolingHandle &cph, const Tensor &dy, const
Tensor& x, const Tensor& y);
 
+
+class CudnnRNNHandle {
+ public:
+   CudnnRNNHandle(const std::vector<Tensor> &x,
+       const int feature_size,
+       const int hidden_size,
+       const int mode = 0,
+       const int num_layers = 1,
+       const int  bias = 1,
+       const float dropout = 0.0f,
+       const int bidirectional = 0);
 
 Review comment:
   num_layers, dropout and bidirectional can be exposed to python?

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services

Mime
View raw message