Searched refs:ReplaceTerminator (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 237 void ReplaceTerminator(Block* block, ArrayRef<Value> extra_results, in ReplaceTerminator() function
263 ReplaceTerminator(&op.then_branch().front(), /*extra_results=*/{}, &builder); in LowerIfRegion()
267 ReplaceTerminator(&op.else_branch().front(), /*extra_results=*/{}, &builder); in LowerIfRegion()
294 ReplaceTerminator(®ion.front(), /*extra_results=*/{}, &builder, in LowerCaseRegion()
346 ReplaceTerminator(&cond_block, /*extra_results=*/{}, &builder, in LowerWhileRegion()
360 ReplaceTerminator(&body_block, implicit_input_elements, &builder); in LowerWhileRegion()
|