-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
Add 'get_num_inputs' to GraphRuntime #6118
Conversation
Update with Apache TVM
just curious, what is it used for? |
The number contains both inputs and weights. Is that what you want? |
Yes, not just the input data but potential input weights as well. I want the length of "arg_nodes" from the graph json. |
I have an app level use case that this information is useful for. Implementation mirrors the existing 'get_num_outputs' API.