From bbc77cc4abaa16d72eb0ac32d9bb10d3e4cd4229 Mon Sep 17 00:00:00 2001 From: Anton Cheshkov Date: Wed, 23 Jul 2025 21:02:58 +0800 Subject: [PATCH 1/3] fixed --- tools/convert_tool/tensorflow/tf2tengine.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tools/convert_tool/tensorflow/tf2tengine.cpp b/tools/convert_tool/tensorflow/tf2tengine.cpp index 3658f9514..df855a097 100644 --- a/tools/convert_tool/tensorflow/tf2tengine.cpp +++ b/tools/convert_tool/tensorflow/tf2tengine.cpp @@ -2489,6 +2489,10 @@ 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,9 +2502,12 @@ 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; } + int load_concat(TFNode* tf_node, TFGraph& tf_graph, ir_graph_t* graph, ir_node_t* node) { struct concat_param* param = (struct concat_param*)node->op.param_mem; From bcbfc39848194052198b6590c684e91aed9ce3fb Mon Sep 17 00:00:00 2001 From: Anton Date: Wed, 3 Sep 2025 15:25:11 +0300 Subject: [PATCH 2/3] removed spaces --- tools/convert_tool/tensorflow/tf2tengine.cpp | 2 -- 1 file changed, 2 deletions(-) diff --git a/tools/convert_tool/tensorflow/tf2tengine.cpp b/tools/convert_tool/tensorflow/tf2tengine.cpp index df855a097..21f85f832 100644 --- a/tools/convert_tool/tensorflow/tf2tengine.cpp +++ b/tools/convert_tool/tensorflow/tf2tengine.cpp @@ -2492,7 +2492,6 @@ int load_pad(TFNode* tf_node, TFGraph& tf_graph, ir_graph_t* graph, ir_node_t* n if (!paddings) { return -1; } - param->mode = 0; param->pad_0_h = paddings[0]; param->pad_0_w = paddings[1]; @@ -2502,7 +2501,6 @@ 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; } From 388b436ab7b682365afdef32dd430fd964da2e3c Mon Sep 17 00:00:00 2001 From: Anton Date: Wed, 3 Sep 2025 15:26:13 +0300 Subject: [PATCH 3/3] remove spaces --- tools/convert_tool/tensorflow/tf2tengine.cpp | 1 - 1 file changed, 1 deletion(-) diff --git a/tools/convert_tool/tensorflow/tf2tengine.cpp b/tools/convert_tool/tensorflow/tf2tengine.cpp index 21f85f832..8bd20b2e2 100644 --- a/tools/convert_tool/tensorflow/tf2tengine.cpp +++ b/tools/convert_tool/tensorflow/tf2tengine.cpp @@ -2505,7 +2505,6 @@ int load_pad(TFNode* tf_node, TFGraph& tf_graph, ir_graph_t* graph, ir_node_t* n return 0; } - int load_concat(TFNode* tf_node, TFGraph& tf_graph, ir_graph_t* graph, ir_node_t* node) { struct concat_param* param = (struct concat_param*)node->op.param_mem;