Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ONNX Graph Surgeon] Sequence/map types are not handled #3986

Open
DeltaEpsilon7787 opened this issue Jul 6, 2024 · 4 comments
Open

[ONNX Graph Surgeon] Sequence/map types are not handled #3986

DeltaEpsilon7787 opened this issue Jul 6, 2024 · 4 comments

Comments

@DeltaEpsilon7787
Copy link

Description

In ONNX graph surgeon, it appears that models whose edges have sequence and, possibly, map type don't get properly imported.

image
image
image

@lix19937
Copy link

lix19937 commented Jul 7, 2024

I suggest that zipmap op need split two output.

@DeltaEpsilon7787
Copy link
Author

The amount of outputs can be arbitrary because it accepts classlabels_int / classlabels_str attribute which can be any length.

@lix19937
Copy link

lix19937 commented Jul 7, 2024

That is your outputs are dynamic shape. TRT support this feature.

@DeltaEpsilon7787
Copy link
Author

Ah, well, the issue is with onnx_graphsurgeon Python library.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants