Searched refs:body_block (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | functional_control_flow_to_cfg.cc | 219 Block* body_block = builder.createBlock(orig_block_tail); in LowerWhileOp() local
230 body_block->addArgument(type); in LowerWhileOp()
252 PrepareValsForJump(loc, get_cond_arg, body_block, &builder); in LowerWhileOp()
253 builder.create<CondBranchOp>(loc, condition, body_block, br_operands, in LowerWhileOp()
258 builder.setInsertionPointToEnd(body_block); in LowerWhileOp()
259 auto get_body_arg = [&](int i) { return body_block->getArgument(i); }; in LowerWhileOp()
|
| /external/python/jinja/tests/ |
| D | test_idtracking.py | 120 body_block = nodes.Block(
147 body_block,
185 body_sym = symbols_for_node(body_block)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 353 Block& body_block = body.front(); in LowerWhileRegion() local
354 builder.setInsertionPointToStart(&body_block); in LowerWhileRegion()
355 ReplaceBlockArgs(&body_block, tuple_input.getType(), &builder); in LowerWhileRegion()
359 &body_block, inputs_size, implicit_inputs.getArrayRef(), &builder); in LowerWhileRegion()
360 ReplaceTerminator(&body_block, implicit_input_elements, &builder); in LowerWhileRegion()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_control_flow.cc | 137 auto* body_block = mapper.lookup(&while_op.body().front()); in LowerWhileOp() local
176 builder.create<mlir::CondBranchOp>(loc, cond_value, body_block, in LowerWhileOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2683 auto &body_block = while_op.body().front(); in matchAndRewrite() local
2685 auto &yield = *body_block.getTerminator(); in matchAndRewrite()
2686 for (auto ba : body_block.getArguments()) { in matchAndRewrite()
2731 if (body_block.getArgument(arg_index).use_empty() && in matchAndRewrite()
2737 body_block.eraseArgument(arg_index); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2986 auto &body_block = while_op.body().front(); in matchAndRewrite() local
2988 auto &yield = *body_block.getTerminator(); in matchAndRewrite()
2994 auto body_arg = body_block.getArgument(op_idx); in matchAndRewrite()
3015 if (body_block.getArgument(op_idx).use_empty() && in matchAndRewrite()
|