-
Notifications
You must be signed in to change notification settings - Fork 2.4k
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
Fix tile shape inference when repeats got dynamic shape #15792
Fix tile shape inference when repeats got dynamic shape #15792
Conversation
const auto r = (repeats_rank.is_static() && repeats_shape[0].is_static()) ? repeats_shape[0].get_length() : 0; | ||
output_shape.resize(std::max<size_t>(arg_rank.get_length(), r)); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is incorrect behavior. Please see the specification. repeats could be longer than data rank and it will extend output rank.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Changed to return dynamic shape.
…kit#15792) * fix shape infer when repeats got dynamic shape * Dynamic output shape when repeats dim is dynamic
Details:
Fix tile shape calculation in case when repeats input has got dynamic shape.