test=develop

7 years ago · 782ef3c5dc
parent 8e8e8e66ab
commit 782ef3c5dc
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -7494,7 +7494,7 @@ def reorg(x, stride, name=None):
        raise ValueError("stride must be a python Int")

    if name is None:
-        out = helper.create_tmp_variable(dtype=x.dtype)
+        out = helper.create_variable_for_type_inference(dtype=x.dtype)
    else:
        out = helper.create_variable(
            name=name, dtype=x.dtype, persistable=False)