diff --git a/x2paddle/op_mapper/dygraph/tf2paddle/tf_op_mapper.py b/x2paddle/op_mapper/dygraph/tf2paddle/tf_op_mapper.py index 88bda2a34f215aa59b4892c931e906867dabf72f..a326e8abf43df44c786d909f9480b7b920701514 100644 --- a/x2paddle/op_mapper/dygraph/tf2paddle/tf_op_mapper.py +++ b/x2paddle/op_mapper/dygraph/tf2paddle/tf_op_mapper.py @@ -921,7 +921,7 @@ class TFOpMapper(OpMapper): # outputs=[reshape_name], # shape=shape) # inputs['offsets'] = reshape_name - begin = self.decoder.infer_tensor(begin, use_diff_inputs=False).to_list() + begin = self.decoder.infer_tensor(begin, use_diff_inputs=False).tolist() attrs['offsets'] = begin if size.layer_type == "Const": size = size.value.tolist() diff --git a/x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py b/x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py index c35f42b434a302d86620001709af6a7c0551e603..851810e246ae2cbb476191e6ac06159eedc0a87f 100644 --- a/x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py +++ b/x2paddle/op_mapper/static/tf2paddle/tf_op_mapper.py @@ -897,7 +897,7 @@ class TFOpMapper(OpMapper): # outputs=[reshape_name], # shape=shape) # inputs['offsets'] = reshape_name - begin = self.decoder.infer_tensor(begin, use_diff_inputs=False).to_list() + begin = self.decoder.infer_tensor(begin, use_diff_inputs=False).tolist() attrs['offsets'] = begin if size.layer_type == "Const": size = size.value.tolist()