-
Notifications
You must be signed in to change notification settings - Fork 1.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
Support cover_all=True
on Unpooling2D
in exporting to ONNX
#8391
Conversation
Jenkins. test this please! |
Jenkins CI test (for commit 69d3cc2, target branch master) succeeded! |
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.
LGTM!
@@ -24,6 +24,11 @@ | |||
'in_shape': (1, 3, 6, 5, 4), 'args': [3, 2, 1], 'cover_all': True}, | |||
{'op_name': 'unpooling_2d', | |||
'in_shape': (1, 3, 6, 6), 'args': [3, None, 0], 'cover_all': False}, | |||
# when cover_all=True, interpolation between Chainer and ONNXRuntime |
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 feel we need a TODO for this.
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.
add TODO, thx
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.
CI is OK so LGTM 👍
Jenkins, test this please! |
Jenkins CI test (for commit 72a2edc, target branch master) succeeded! |
cover_all=True
on Unpooling2D
in exporting to ONNX
Jenkins CI test (for commit 72a2edc, target branch master) succeeded! |
Support
cover_all=True
on Chainer'sUnpool2d
. Simply recalculate output size. Interpolation logic is differ between Chainer and ONNXRuntime, and output values checking is skipped.