-
Notifications
You must be signed in to change notification settings - Fork 74k
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
[MLIR] Add folder for mhlo get_dimension_size #42509
[MLIR] Add folder for mhlo get_dimension_size #42509
Conversation
b70e0d9
to
d150a91
Compare
d150a91
to
ae581ca
Compare
ae581ca
to
4969511
Compare
Fixed dialect name in test cases: xla_hlo -> mhlo |
@sherhut @joker-eph The internal builds here may fail here since a lot of ops would get folded away (since shapes are typically constant, and more operands would become constant with this folding). |
@bondhugula can you please fix ubuntu sanity build failures ? |
Seems like in conflicts, can you rebase? |
Done. |
4969511
to
90fef6d
Compare
Add folder for mhlo GetDimensionSizeOp. get_dimension_size folds to a constant when the corresponding tensor dimension size is statically known / constant.
Rebased and fixed conflict. |
90fef6d
to
5994915
Compare
Imported from GitHub PR tensorflow/tensorflow#42509 Add folder for mhlo GetDimensionSizeOp (`mhlo.get_dimension_size`). `get_dimension_size` folds to a constant when the corresponding tensor dimension size is statically known / constant. Copybara import of the project: -- 5994915525ec2e932125aa1f133ce2260ba100af by Uday Bondhugula <uday@polymagelabs.com>: [MLIR] Add folder for mhlo get_dimension_size Add folder for mhlo GetDimensionSizeOp. get_dimension_size folds to a constant when the corresponding tensor dimension size is statically known / constant. COPYBARA_INTEGRATE_REVIEW=tensorflow/tensorflow#42509 from polymage-labs:get_dimension_size_fold 5994915525ec2e932125aa1f133ce2260ba100af PiperOrigin-RevId: 328222517
Add folder for mhlo GetDimensionSizeOp (
mhlo.get_dimension_size
).get_dimension_size
folds to a constant when the corresponding tensordimension size is statically known / constant.