Searched refs:region_while (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 644 void RewriteRegionWhileOp(OpBuilder& builder, WhileOp region_while, in RewriteRegionWhileOp() argument
651 GetValueWithToken(builder, region_while.val(), token, rewritten_operands); in RewriteRegionWhileOp()
653 auto new_result_type = GetTypeWithToken(builder, region_while.getType()); in RewriteRegionWhileOp()
656 auto new_while = builder.create<WhileOp>(region_while.getLoc(), in RewriteRegionWhileOp()
660 new_while.cond().takeBody(region_while.cond()); in RewriteRegionWhileOp()
661 new_while.body().takeBody(region_while.body()); in RewriteRegionWhileOp()
665 ReplaceWithTupleResult(builder, region_while.getResult(), in RewriteRegionWhileOp()
672 region_while.erase(); in RewriteRegionWhileOp()
689 OpBuilder& builder, WhileOp region_while, Optional<unsigned> region_idx, in ProcessRegionWhileOp() argument
692 builder.setInsertionPoint(region_while); in ProcessRegionWhileOp()
[all …]
|