Searched refs:ImportXlaRegion (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 74 void ImportXlaRegion(mlir::FuncOp func, Region* dest_region, Location loc, in ImportXlaRegion() function
115 ImportXlaRegion(op.then_function(), &if_op.true_branch(), loc); in LowerIf()
116 ImportXlaRegion(op.else_function(), &if_op.false_branch(), loc); in LowerIf()
143 ImportXlaRegion(branch_func, &case_op.branches()[i], loc, in LowerCase()
167 ImportXlaRegion(op.body_function(), &while_op.body(), loc); in LowerWhile()
168 ImportXlaRegion(op.cond_function(), &while_op.cond(), loc, in LowerWhile()
|