From 0a40bebbd53866aee206f411cfb1c25ca15a916a Mon Sep 17 00:00:00 2001 From: jiangjiajun Date: Wed, 12 Aug 2020 03:21:07 +0000 Subject: [PATCH] cast dtype for onnx2paddle: Tile --- x2paddle/op_mapper/onnx2paddle/opset9/opset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/x2paddle/op_mapper/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/onnx2paddle/opset9/opset.py index 4bbaa13..c08be94 100644 --- 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): -- GitLab