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

[infershape] should be able to set tensor shapes - inputs and others #22

Open
zhenhuaw-me opened this issue Aug 25, 2022 · 0 comments
Open
Labels
enhancement New feature or request

Comments

@zhenhuaw-me
Copy link
Owner

infershape is not very useful if the input shapes are symbolics (dynamic shapes). If the user can set input shapes, it's more powerful:

  • If set to static shapes, the shape of the model will be known.
  • Even for symbolics, the user can update the input shapes.

The setup should be optional, and can extend to all the tensors in the model (excluding shape op related).

Interface should be something like below.

onnx infershape path/to/input/model.onnx path/to/output/model.onnx --tensor-shape t1:[d0,d1] t2:[d0,d1,d3]
@zhenhuaw-me zhenhuaw-me added the enhancement New feature or request label Aug 25, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant