From daad6251215edfebdfd27e6230a044827d0fb467 Mon Sep 17 00:00:00 2001
From: liu zhengxi <380185688@qq.com>
Date: Sat, 15 Aug 2020 13:57:25 +0800
Subject: [PATCH] fix rnncell errmsg (#26298)

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

diff --git a/python/paddle/fluid/layers/rnn.py b/python/paddle/fluid/layers/rnn.py
index ecc5876852..39d25a9c7f 100644
--- a/python/paddle/fluid/layers/rnn.py
+++ b/python/paddle/fluid/layers/rnn.py
@@ -127,7 +127,8 @@ class RNNCell(object):
         else:
             integer_types = (int, )
         check_variable_and_dtype(batch_ref, 'batch_ref',
-                                 ['float32', 'float64'], 'RNNCell')
+                                 ['float32', 'float64', 'int32', 'int64'],
+                                 'RNNCell')
         check_type(shape, 'shape', (list, tuple, type(None), integer_types),
                    'RNNCell')
         if isinstance(shape, (list, tuple)):
-- 
GitLab