Searched refs:cond_block (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_control_flow.cc | 136 auto* cond_block = mapper.lookup(&while_op.cond().front()); in LowerWhileOp() local
143 builder.create<mlir::BranchOp>(loc, cond_block, while_op.getOperand()); in LowerWhileOp()
156 builder.setInsertionPointToStart(cond_block); in LowerWhileOp()
174 llvm::SmallVector<Value, 4> successor_args(cond_block->args_begin(), in LowerWhileOp()
175 cond_block->args_end()); in LowerWhileOp()
197 builder.create<mlir::BranchOp>(loc, cond_block, return_op.getOperands()); in LowerWhileOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 218 Block* cond_block = builder.createBlock(orig_block_tail); in LowerWhileOp() local
227 cond_block->addArgument(type); in LowerWhileOp()
239 JumpToBlock(loc, get_operand, cond_block, &builder); in LowerWhileOp()
244 builder.setInsertionPointToEnd(cond_block); in LowerWhileOp()
246 auto get_cond_arg = [&](int i) { return cond_block->getArgument(i); }; in LowerWhileOp()
263 JumpToBlock(loc, get_body_result, cond_block, &builder); in LowerWhileOp()
|
| /external/llvm-project/llvm/test/Transforms/RewriteStatepointsForGC/ |
| D | check_traversal_order.ll | 25 br i1 %cond, label %cond_block, label %exit
27 cond_block:
31 %tmp4 = phi i8 addrspace(1)* [ %tmp2, %loop ], [ %tmp2, %cond_block ]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 340 Block& cond_block = cond.front(); in LowerWhileRegion() local
341 builder.setInsertionPointToStart(&cond_block); in LowerWhileRegion()
342 ReplaceBlockArgs(&cond_block, tuple_input.getType(), &builder); in LowerWhileRegion()
343 ReplaceImplicitInputs(&cond_block, inputs_size, implicit_inputs.getArrayRef(), in LowerWhileRegion()
346 ReplaceTerminator(&cond_block, /*extra_results=*/{}, &builder, in LowerWhileRegion()
|
| /external/selinux/libsepol/cil/src/ |
| D | cil_verify.c | 967 struct cil_tree_node *cond_block = node->cl_head; in __cil_verify_booleanif() local
969 while (cond_block != NULL) { in __cil_verify_booleanif()
970 rc = cil_tree_walk(cond_block, __cil_verify_booleanif_helper, NULL, NULL, symtab); in __cil_verify_booleanif()
974 cond_block = cond_block->next; in __cil_verify_booleanif()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2684 auto &cond_block = while_op.cond().front(); in matchAndRewrite() local
2708 cond_block.getArgument(arg_no).replaceAllUsesWith(value); in matchAndRewrite()
2732 cond_block.getArgument(arg_index).use_empty() && in matchAndRewrite()
2738 cond_block.eraseArgument(arg_index); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2987 auto &cond_block = while_op.cond().front(); in matchAndRewrite() local
3004 auto cond_arg = cond_block.getArgument(op_idx); in matchAndRewrite()
3016 cond_block.getArgument(op_idx).use_empty() && in matchAndRewrite()
|