-
Notifications
You must be signed in to change notification settings - Fork 19.6k
[OpenVINO backend] support repeat #21433
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?
[OpenVINO backend] support repeat #21433
Conversation
3ff9ca3
to
8dffc46
Compare
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## master #21433 +/- ##
==========================================
- Coverage 82.72% 82.72% -0.01%
==========================================
Files 565 565
Lines 55219 55287 +68
Branches 8608 8617 +9
==========================================
+ Hits 45682 45735 +53
- Misses 7427 7443 +16
+ Partials 2110 2109 -1
Flags with carried forward coverage won't be shown. Click here to find out more. ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
f4ce2c6
to
345f77c
Compare
345f77c
to
aefa32f
Compare
@@ -1291,7 +1291,96 @@ def reciprocal(x): | |||
|
|||
|
|||
def repeat(x, repeats, axis=None): | |||
raise NotImplementedError("`repeat` is not supported with openvino backend") | |||
x = get_ov_output(x) |
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.
I think you can use Tile
operation from our opset. It should be much easier to express np.repeats
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.
tile
repeats the whole tensor, but repeat needs to repeat each element in a tensor, so I am using tile
for scalar repeating and using range
, cumsum
, and gather
for tensor's repeating.
@rkazants