diff --git a/tools/convert_tool/tensorflow/tf2tengine.cpp b/tools/convert_tool/tensorflow/tf2tengine.cpp index 3658f9514..8bd20b2e2 100644 --- a/tools/convert_tool/tensorflow/tf2tengine.cpp +++ b/tools/convert_tool/tensorflow/tf2tengine.cpp @@ -2489,6 +2489,9 @@ int load_pad(TFNode* tf_node, TFGraph& tf_graph, ir_graph_t* graph, ir_node_t* n TFNode* input = tf_node->inputs[1]; int* paddings = (int*)LoadConstParam(input); + if (!paddings) { + return -1; + } param->mode = 0; param->pad_0_h = paddings[0]; param->pad_0_w = paddings[1]; @@ -2498,6 +2501,7 @@ int load_pad(TFNode* tf_node, TFGraph& tf_graph, ir_graph_t* graph, ir_node_t* n param->pad_2_w = paddings[5]; param->pad_3_h = paddings[6]; param->pad_3_w = paddings[7]; + free(paddings); return 0; }