-
Notifications
You must be signed in to change notification settings - Fork 315
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
Handle more general cases in ONNX.Expand
lowering to stablehlo
#2747
Handle more general cases in ONNX.Expand
lowering to stablehlo
#2747
Conversation
Signed-off-by: Sam <[email protected]>
Signed-off-by: Sam <[email protected]>
Signed-off-by: Sam <[email protected]>
Can one of the admins verify this patch? |
Signed-off-by: Sam <[email protected]>
Can one of the admins verify this patch? |
Can one of the admins verify this patch? |
Can one of the admins verify this patch? |
Can one of the admins verify this patch? |
@tungld can i get a review on 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.
LGTM. Thanks!
Jenkins Linux amd64 Build #14386 [push] Handle more general case... started at 04:20 |
Jenkins Linux ppc64le Build #13411 [push] Handle more general case... started at 05:29 |
Jenkins Linux s390x Build #14416 [push] Handle more general case... started at 05:20 |
Jenkins Linux amd64 Build #14386 [push] Handle more general case... passed after 1 hr 3 min |
Jenkins Linux s390x Build #14416 [push] Handle more general case... passed after 1 hr 22 min |
Jenkins Linux ppc64le Build #13411 [push] Handle more general case... passed after 1 hr 57 min |
Previously the lowering only worked if the defining op of the
shape
operand was a constant orONNX.Shape
op. It is unnecessary to specialize for the latter, so this patch generalizes it.