-
Notifications
You must be signed in to change notification settings - Fork 2.7k
MultiScaleDeformableAttn implementation and optimization #31191
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
base: master
Are you sure you want to change the base?
MultiScaleDeformableAttn implementation and optimization #31191
Conversation
//({flatten_Slice_1194, {-1}}, {{"axis", 0}}); | ||
// ({Unsqueeze_65524 | Unsqueeze_28998, Unsqueeze_65525 | Unsqueeze_28999}, | ||
// wrap_type<opset1::Concat>(pattern::consumers_count(1)); | ||
auto attn_Concat_17 = wrap_type<Concat>(grid_sampler_block, {{"axis", -2}}); |
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.
@itikhono any good way to describe the pattern of a flexible number of Concat branches?
I am trying this way by passing one grid_sampler_block to it only, but the pattern check at this place fails due to the number of inputs of Concat do not match.
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.
1dc7309
to
51ad086
Compare
3969041
to
9b8e88d
Compare
be512dc
to
4ec8009
Compare
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.
Can be PR cleaned as it base other PRs some changes should not be visible as diff
* @brief The transformation to support multi_scale_deformable_attn | ||
* \ingroup ov_pass_cpp_api | ||
*/ | ||
class OPENVINO_API MultiScaleDeformableAttn : public ModelPass { |
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.
Why this class in public API not in transformation component?
Details:
It is based on #30587
Tickets: