-
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
[PyOV] Extend Python API with Col2Im-15
#24569
Conversation
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.
Overall LGTM.
for param in [strides, dilations]: | ||
if param is None: | ||
param = [1, 1] | ||
for param in [pads_begin, pads_end]: | ||
if param is None: | ||
param = [0, 0] |
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 you test these None
cases somehow?
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.
They are tested in the first test case: ((3, 4), [2, 2], [1, 1], [3, 2, 2], []),
. The last empty list is params
, which are given into the op constructor as *params
.
@@ -6,4 +6,5 @@ | |||
|
|||
# TODO (ticket 138273): Add previous opset operators at the end of opset15 development | |||
from openvino.runtime.opset1.ops import parameter | |||
from openvino.runtime.opset15.ops import col2im |
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.
do we need from openvino.runtime.opset1.ops import parameter
above?
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'm not certain, my guess is that it was needed in opset15
for now for the tests to use it. One way or another since this PR is targeted for Code Freeze let's come back to it another time.
### Details: - Similar in functionality to https://pytorch.org/docs/stable/generated/torch.nn.Fold.html, Col2Im is `torch.nn.Fold` restricted to two output spatial dimensions ### Tickets: - CVS-138919 ### Related PRs: - #24197 - #23947 - #24569 --------- Co-authored-by: Michal Lukaszewski <[email protected]>
### Details: - Register `Col2im-15` in CPU Plugin - Use reference implementation, used #23844 as reference - Add tests ### Tickets: - CVS-142438 ### Related PRs: - #24548 - #24197 - #23947 - #24569 --------- Co-authored-by: Michal Lukaszewski <[email protected]> Co-authored-by: Maksim Kutakov <[email protected]>
Details:
Col2Im-15
Col2Im-15
reference implementation #24548 to work and pass CITickets:
Related PRs:
Col2Im-15
reference implementation #24548Col2Im-15
implementation to Core #24197