Skip to content

Commit 0d71be8

Browse files
Filter structured inputs to only include TensorSpecs (#1490)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 4fc6620 commit 0d71be8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tf2onnx/convert.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,8 @@ def main():
273273

274274
def tensor_names_from_structed(concrete_func, input_names, output_names):
275275
tensors_to_rename = {}
276-
structured_inputs = [t.name for t in tf.nest.flatten(concrete_func.structured_input_signature)]
276+
flat_structured_inp = tf.nest.flatten(concrete_func.structured_input_signature)
277+
structured_inputs = [t.name for t in flat_structured_inp if isinstance(t, tf.TensorSpec)]
277278
tensors_to_rename.update(zip(input_names, structured_inputs))
278279
if isinstance(concrete_func.structured_outputs, dict):
279280
for k, v in concrete_func.structured_outputs.items():

0 commit comments

Comments
 (0)