提交 fe75f532 编写于 作者: C Channingss

Merge remote-tracking branch 'paddle/develop' into Conv

......@@ -1338,6 +1338,14 @@ class OpSet9():
if repeats is None:
repeats = val_repeats.layer_name
if val_repeats.dtype != 'int32':
attr = {"dtype": string("int32")}
node.fluid_code.add_layer(
"cast", inputs=repeats,
output="{}.tmp".format(repeats),
param_attr=attr)
repeats = "{}.tmp".format(repeats)
elif isinstance(repeats, int):
repeats = [repeats]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册