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

[luci/service] Fix dynamic shape inference for reshape operation #14061

Merged
merged 1 commit into from
Sep 24, 2024

Conversation

jongwonyang
Copy link
Member

@jongwonyang jongwonyang commented Sep 24, 2024

This commit prevents shape inference when the reshape tensor's dimensions are not fully known.

Related: #13927
Draft: #13999

ONE-DCO-1.0-Signed-off-by: Jongwon Yang [email protected]

@jongwonyang
Copy link
Member Author

jongwonyang commented Sep 24, 2024

WHAT

When I optimize the following circle:
image

I get this which has wrong output shape:
image

WHY

The original code always infers -1 even if the input tensor is not fully known.

HOW

Infer -1 only when the input tensor is fully known.

@jongwonyang jongwonyang requested a review from a team September 24, 2024 02:04
@jongwonyang jongwonyang marked this pull request as ready for review September 24, 2024 03:24
@jongwonyang jongwonyang requested a review from a team September 24, 2024 03:24
@jongwonyang jongwonyang added the PR/ready for review It is ready to review. Please review it. label Sep 24, 2024
This commit prevents shape inference when the reshape tensor's dimensions are not fully known.

ONE-DCO-1.0-Signed-off-by: Jongwon Yang <[email protected]>
Copy link
Contributor

@seanshpark seanshpark left a comment

Choose a reason for hiding this comment

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

LGTM thank you!

Copy link
Contributor

@zetwhite zetwhite left a comment

Choose a reason for hiding this comment

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

LGTM

@seanshpark seanshpark merged commit 8958dc0 into Samsung:master Sep 24, 2024
10 checks passed
Copy link
Contributor

@shs-park shs-park left a comment

Choose a reason for hiding this comment

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

👍

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
PR/ready for review It is ready to review. Please review it. SSAFY
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants