Skip to content

Commit b4ac342

Browse files
Merge pull request #1110 from onnx/tom/FixConcFuncMultIn
Fix --concrete_func for multiple inputs in signature
2 parents 1650716 + 374476f commit b4ac342

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tf2onnx/tf_loader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -290,8 +290,8 @@ def _from_saved_model_v2(model_path, input_names, output_names, tag, signature_d
290290
utils.make_sure(hasattr(imported, "__call__"), err_no_call)
291291
utils.make_sure(concrete_function_index < len(imported.__call__.concrete_functions),
292292
err_index, concrete_function_index, len(imported.__call__.concrete_functions) - 1)
293-
sig = imported.__call__.concrete_functions[concrete_function_index].structured_input_signature[0][0]
294-
concrete_func = imported.__call__.get_concrete_function(sig)
293+
sig = imported.__call__.concrete_functions[concrete_function_index].structured_input_signature[0]
294+
concrete_func = imported.__call__.get_concrete_function(*sig)
295295
elif signature_def:
296296
utils.make_sure(signature_def[0] in valid_sigs, err_sig_nomatch, signature_def[0])
297297
concrete_func = imported.signatures[signature_def[0]]

0 commit comments

Comments
 (0)