From 084ff6574d8122afa256651aec7abb55260220f0 Mon Sep 17 00:00:00 2001
From: phlrain <phliuhongyu@126.com>
Date: Tue, 27 Nov 2018 16:40:26 +0800
Subject: [PATCH] add cudnn lstm; test=develop

---
 python/paddle/fluid/layers/nn.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index ef46bbb2717..d2df819c085 100644
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -525,6 +525,9 @@ def cudnn_lstm(input,
         default_initializer(Initialize|None): Where use initializer to initialize the Weight
                          If set None, defaule initializer will be used
 
+        fix_seed(bool): If it's True, fix seed will used for dropout in LSTM
+        seed(int): If fix_seed is True, dropout seed in LSTM will use this seed 
+
 
     Returns:
         rnn_out(Tensor): result of LSTM hidden, shape is (seq_len x batch_size x hidden_size)
-- 
GitLab