未验证 提交 f745d6d9 编写于 作者: H hong 提交者: GitHub

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

上级 02c370c3
...@@ -9690,7 +9690,8 @@ def expand(x, expand_times, name=None): ...@@ -9690,7 +9690,8 @@ def expand(x, expand_times, name=None):
new_expand_times.append(ele) new_expand_times.append(ele)
else: else:
assert (isinstance(ele, int)) assert (isinstance(ele, int))
temp_out = helper.create_variable_for_type_inference(dtype) temp_out = helper.create_variable_for_type_inference(
"int32")
fill_constant( fill_constant(
[1], 'int32', ele, force_cpu=True, out=temp_out) [1], 'int32', ele, force_cpu=True, out=temp_out)
new_expand_times.append(temp_out) new_expand_times.append(temp_out)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册