diff --git a/x2paddle/op_mapper/tf_op_mapper_nhwc.py b/x2paddle/op_mapper/tf_op_mapper_nhwc.py index a005c27c38cc2de544a7ecfed4c3a4033539c38c..75ab10e2980272308f0615d202a284e1ce6bba96 100644 --- a/x2paddle/op_mapper/tf_op_mapper_nhwc.py +++ b/x2paddle/op_mapper/tf_op_mapper_nhwc.py @@ -744,13 +744,12 @@ class TFOpMapperNHWC(OpMapper): "start": start, "end": limit, "step": delta, - "dtype": string(dtype) } attr = {"dtype": string(node.dtype)} node.fluid_code.add_layer("range", inputs=inputs, output=node, - param_attr=None) + param_attr=attr) def Mean(self, node): input = self.graph.get_node(node.layer.input[0], copy=True)