Change the way to set node name. (#2278)

Signed-off-by: Jay Zhang <[email protected]>
onnx · Dec 15, 2023 · 85a10b5 · 85a10b5
1 parent 07dfaa8
commit 85a10b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tf2onnx/tflite_utils.py b/tf2onnx/tflite_utils.py
@@ -483,6 +483,8 @@ def shape_to_tf_shape(dims):
         if has_prequantized_output:
             output_names = [get_prequant(out) for out in output_names]
         onnx_node = helper.make_node(optype, input_names, output_names, name=output_names[0], **attr)
+        node_name = output_names[0] if output_names else utils.make_name(f"{optype}_Output")
+        onnx_node = helper.make_node(optype, input_names, output_names, name=node_name, **attr)
         onnx_nodes.append(onnx_node)
 
     inputs = [tensor_names[tflite_g.Inputs(i)] for i in range(tflite_g.InputsLength())]