fix expand op dtype build bugs; test=develop (#18932)

6 years ago · f745d6d9e4
parent 02c370c3dc
commit f745d6d9e4
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -9690,7 +9690,8 @@ def expand(x, expand_times, name=None):
                    new_expand_times.append(ele)
                else:
                    assert (isinstance(ele, int))
-                    temp_out = helper.create_variable_for_type_inference(dtype)
+                    temp_out = helper.create_variable_for_type_inference(
+                        "int32")
                    fill_constant(
                        [1], 'int32', ele, force_cpu=True, out=temp_out)
                    new_expand_times.append(temp_out)