cast dtype for onnx2paddle: Tile

0a40bebb · jiangjiajun · 665937e3 · 0a40bebb
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

x2paddle/op_mapper/onnx2paddle/opset9/opset.py x2paddle/op_mapper/onnx2paddle/opset9/opset.py +3 -1

未找到文件。
--- a/x2paddle/op_mapper/onnx2paddle/opset9/opset.py
+++ b/x2paddle/op_mapper/onnx2paddle/opset9/opset.py
@@ -1339,7 +1339,9 @@ class OpSet9():
            if val_repeats.dtype != 'int32':
                attr = {"dtype": string("int32")}
                node.fluid_code.add_layer(
-                    "cast", inputs=repeats, output="{}.tmp".format(repeats),  param_attr=attr)
+                    "cast", inputs=repeats, 
+                    output="{}.tmp".format(repeats),
+                    param_attr=attr)
                repeats = "{}.tmp".format(repeats)

        elif isinstance(repeats, int):