Searched refs:tail_block (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_control_flow.cc | 67 auto* tail_block = orig_block->splitBlock(op_inst); in LowerIfOp() local
74 Region::iterator(tail_block), mapper); in LowerIfOp()
76 Region::iterator(tail_block), mapper); in LowerIfOp()
94 if (failed(ReplaceTerminators(&if_op.true_branch(), tail_block, loc, mapper, in LowerIfOp()
97 if (failed(ReplaceTerminators(&if_op.false_branch(), tail_block, loc, mapper, in LowerIfOp()
101 tail_block->addArguments(if_op.getResult().getType()); in LowerIfOp()
102 if_op.getResult().replaceAllUsesWith(tail_block->getArgument(0)); in LowerIfOp()
127 auto* tail_block = orig_block->splitBlock(op_inst); in LowerWhileOp() local
131 Region::iterator(tail_block), mapper); in LowerWhileOp()
133 Region::iterator(tail_block), mapper); in LowerWhileOp()
[all …]
|