fix test_layers, test=develop

PaddlePaddle · Aug 13, 2020 · a2a22f5 · a2a22f5
1 parent f6ac45c
commit a2a22f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
@@ -7475,7 +7475,8 @@ def _is_list_or_turple_(data):
         elif isinstance(scale, float) or isinstance(scale, int):
             if scale <= 0:
                 raise ValueError("Attr(scale) should be greater than zero.")
-            attrs['scale'] = float(scale)
+            attrs['scale_w'] = attrs['scale_h'] = attrs['scale_d'] = float(
+                scale)
         else:
             raise TypeError(
                 "Attr(scale)'s type should be float, int or Variable.")