fix code. test=develop

Change-Id: Ia9ce2b225e04171e9bf248f4965d28e760f785bf
PaddlePaddle · Aug 18, 2020 · 3799723 · 3799723
1 parent e52cb39
commit 3799723
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_reshape_op.py b/python/paddle/fluid/tests/unittests/test_reshape_op.py
@@ -286,7 +286,7 @@ def test_imperative(self):
         self._set_paddle_api()
         input = np.random.random([2, 25]).astype("float32")
         shape = [2, 5, 5]
-        with paddle.imperative.guard():
+        with fluid.dygraph.guard():
             x = self.to_tensor(input)
             positive_five = self.fill_constant([1], "int32", 5)
 

diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
@@ -376,7 +376,7 @@ def roll(x, shifts, axis=None, name=None):
         outputs={'Out': out},
         attrs={'axis': axis,
                'shifts': shifts})
-    out = reshape(out, shape=origin_shape, inplace=True)
+    out = layers.reshape(out, shape=origin_shape, inplace=True)
     return out