-
Notifications
You must be signed in to change notification settings - Fork 157
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
[compiler] Support dynamic shape inference for Conv2D and DepthwiseConv2D #14115
Comments
So I am planning this in the following sequence:
However, I am concerned about the migration process.
There is a function infer_conv2d_type() that is shared by Conv2D and DepthwiseConv2D.To migrate this, I have considered several ways to do this:
Which option do you prefer? |
Please go one with your first method and lets see how it comes out. |
Thank you. I'll work on this way. |
Currently, we are working on enabling Yamnet to support dynamic shape inference for the Reshape operation.
In this process, we planned to infer all dimensions as unknown when the shape is not CircleConst.
According to our decision, the height, width, and number of channels will be provided to Conv2D in unknown state, resulting an incorrect inference (the unknown dim will be treated as
0
).Therefore, if the height, width, and channels of the IFM are unknown, Conv2D should infer those dimensions as unknown.
The text was updated successfully, but these errors were encountered: