Home
last modified time | relevance | path

Searched refs:new_if_op (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmerge_control_flow.cc218 auto new_if_op = builder.create<TF::IfRegionOp>( in CreateMergedIf() local
222 new_if_op.then_branch().push_back(new Block); in CreateMergedIf()
223 new_if_op.else_branch().push_back(new Block); in CreateMergedIf()
238 MoveResultsAfter(destination.getOperation(), new_if_op.getOperation(), in CreateMergedIf()
245 new_if_op.getResult(new_return_index++)); in CreateMergedIf()
249 new_if_op.getResult(new_return_index++)); in CreateMergedIf()
260 builder.setInsertionPointToEnd(&new_if_op.then_branch().front()); in CreateMergedIf()
272 builder.setInsertionPointToEnd(&new_if_op.else_branch().front()); in CreateMergedIf()
278 MoveBranches(/*source=*/destination, /*destination=*/new_if_op); in CreateMergedIf()
280 MoveBranches(/*source=*/source, /*destination=*/new_if_op); in CreateMergedIf()
[all …]