diff --git a/python/paddle/fluid/communicator.py b/python/paddle/fluid/communicator.py
index 814a70a10e06cf5867a9ed9b736a895f50f24a49..b203e2a80bda49d54e3ed9a4a7cd616cf4fc7b3e 100644
--- a/python/paddle/fluid/communicator.py
+++ b/python/paddle/fluid/communicator.py
@@ -162,3 +162,6 @@ class LargeScaleKV(object):
 
     def load(self, varname, dirname):
         self.scale_kv.load(varname, dirname)
+
+    def size(self, varname):
+        return self.scale_kv.size(varname)
diff --git a/python/paddle/static/nn/__init__.py b/python/paddle/static/nn/__init__.py
index 91da0926b1870bb4a7999e62965c135dcf36bf25..0c5d42ad1a1ecf5ba7f4668b0046ab94e61e5494 100644
--- a/python/paddle/static/nn/__init__.py
+++ b/python/paddle/static/nn/__init__.py
@@ -34,6 +34,7 @@ __all__ = [
     'prelu',
     'row_conv',
     'spectral_norm',
+    'reshape',
 ]
 
 from ...fluid.layers import fc  #DEFINE_ALIAS
@@ -58,3 +59,4 @@ from ...fluid.layers import row_conv  #DEFINE_ALIAS
 from ...fluid.layers import spectral_norm  #DEFINE_ALIAS
 
 from ...fluid.input import embedding  #DEFINE_ALIAS
+from ...fluid.layers import reshape  #DEFINE_ALIAS