We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4fc6620 commit 0d71be8Copy full SHA for 0d71be8
tf2onnx/convert.py
@@ -273,7 +273,8 @@ def main():
273
274
def tensor_names_from_structed(concrete_func, input_names, output_names):
275
tensors_to_rename = {}
276
- structured_inputs = [t.name for t in tf.nest.flatten(concrete_func.structured_input_signature)]
+ 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)]
278
tensors_to_rename.update(zip(input_names, structured_inputs))
279
if isinstance(concrete_func.structured_outputs, dict):
280
for k, v in concrete_func.structured_outputs.items():
0 commit comments