Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_control_flow.cc66 auto orig_block = op_inst->getBlock(); in LowerIfOp() local
67 auto* tail_block = orig_block->splitBlock(op_inst); in LowerIfOp()
73 if_op.true_branch().cloneInto(orig_block->getParent(), in LowerIfOp()
75 if_op.false_branch().cloneInto(orig_block->getParent(), in LowerIfOp()
83 builder.setInsertionPointToEnd(orig_block); in LowerIfOp()
126 auto* orig_block = op_inst->getBlock(); in LowerWhileOp() local
127 auto* tail_block = orig_block->splitBlock(op_inst); in LowerWhileOp()
130 while_op.cond().cloneInto(orig_block->getParent(), in LowerWhileOp()
132 while_op.body().cloneInto(orig_block->getParent(), in LowerWhileOp()
142 builder.setInsertionPointToEnd(orig_block); in LowerWhileOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_cfg.cc146 Block* orig_block = op_inst->getBlock(); in LowerIfOp() local
147 Block* merge_block = orig_block->splitBlock(op); in LowerIfOp()
175 builder.setInsertionPointToEnd(orig_block); in LowerIfOp()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dloop_fusion.cpp522 void ReplacePhiParentWith(Instruction* inst, uint32_t orig_block, in ReplacePhiParentWith() argument
524 if (inst->GetSingleWordInOperand(1) == orig_block) { in ReplacePhiParentWith()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_fusion.cpp522 void ReplacePhiParentWith(Instruction* inst, uint32_t orig_block, in ReplacePhiParentWith() argument
524 if (inst->GetSingleWordInOperand(1) == orig_block) { in ReplacePhiParentWith()