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

[TFLite] Added ability to infer shapes for arguments #7293

Merged
merged 1 commit into from
Jan 20, 2021

Conversation

d-smirnov
Copy link
Contributor

@d-smirnov d-smirnov commented Jan 15, 2021

This PR adds an ability to infer argument shapes if shapes are not present in
TFLite files. The set of networks on which the patch was tested is
internal to Arm. Any help with creating unit tests would be appreciated.
I wanted to create a test of two - three operations like this:
image
However was unable to produce tflite (using Keras) file without output shapes. Would like to hear advises.

Added an ability to infer argument shapes if shapes are not present in
TFLite files. The set of networks on which the patch was tested is
internal to Arm. Any help with creating unit tests would be appreciated.
@d-smirnov
Copy link
Contributor Author

cc @anijain2305

Copy link
Contributor

@jwfromm jwfromm left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This looks like a clear improvement to me, thanks for making the change.

@anijain2305 anijain2305 merged commit f8c55db into apache:main Jan 20, 2021
@anijain2305
Copy link
Contributor

Thanks @d-smirnov @jwfromm

TusharKanekiDey pushed a commit to TusharKanekiDey/tvm that referenced this pull request Jan 20, 2021
Added an ability to infer argument shapes if shapes are not present in
TFLite files. The set of networks on which the patch was tested is
internal to Arm. Any help with creating unit tests would be appreciated.
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Jan 21, 2021
Added an ability to infer argument shapes if shapes are not present in
TFLite files. The set of networks on which the patch was tested is
internal to Arm. Any help with creating unit tests would be appreciated.
electriclilies pushed a commit to electriclilies/tvm that referenced this pull request Feb 18, 2021
Added an ability to infer argument shapes if shapes are not present in
TFLite files. The set of networks on which the patch was tested is
internal to Arm. Any help with creating unit tests would be appreciated.
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

Successfully merging this pull request may close these issues.

None yet

3 participants