Skip to content
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

[tf-frontend] lowering tf.if -> tf.if_region -> scf.if #345

Merged
merged 1 commit into from
Jun 18, 2024
Merged

Conversation

heromapwrd
Copy link
Collaborator

  1. lowering tf.if -> tf.if_region -> scf.if
    a. add mlir::TF::CreateTFFunctionalControlFlowToRegions pass to lowering tf.if -> tf.if_region
    b. add mlir::TF::createConvertTfControlFlowToScfPass pass to lowering tf.if_region -> scf.if

  2. inline func call in scf.if
    a. add mlir::tfext::createInlineFuncCallInScfIfPass to inline func::CallOp in scf.if

  3. convert tf.Tile to mhlo op
    a. add ConvertTileOp pattern to convert tf.Tile to mhlo op, when right operand of tf.Tile is dynamic

  4. convert scf.if to mhlo.if
    a. add ConvertScfIfOp pattern to convert scf.if to mhlo.if, but not applied

@qingyunqu qingyunqu merged commit 3739b4d into main Jun 18, 2024
3 checks passed
@qingyunqu qingyunqu deleted the tf-frontend branch June 18, 2024 15:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants