Home
last modified time | relevance | path

Searched refs:if_data (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.cc314 llvm_ir::LlvmIfData if_data; in EmitIfThenElse() local
315 if_data.if_block = b->GetInsertBlock(); in EmitIfThenElse()
316 if_data.true_block = in EmitIfThenElse()
318 if_data.false_block = in EmitIfThenElse()
323 if (if_data.if_block->getTerminator() == nullptr) { in EmitIfThenElse()
324 b->SetInsertPoint(if_data.if_block); in EmitIfThenElse()
325 if_data.after_block = in EmitIfThenElse()
327 b->CreateBr(if_data.after_block); in EmitIfThenElse()
329 if_data.after_block = if_data.if_block->splitBasicBlock( in EmitIfThenElse()
335 if_data.if_block->getTerminator()->eraseFromParent(); in EmitIfThenElse()
[all …]
Dkernel_support_library.cc63 llvm_ir::LlvmIfData if_data = llvm_ir::EmitIfThenElse(condition, name, b_); in IfWithStatus() local
64 b_->SetInsertPoint(&if_data.true_block->back()); in IfWithStatus()
66 b_->SetInsertPoint(&if_data.false_block->back()); in IfWithStatus()
68 llvm_ir::SetToLastInsertPoint(if_data.after_block, b_); in IfWithStatus()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc940 llvm_ir::LlvmIfData if_data = llvm_ir::EmitIfThenElse( in EmitErfInv() local
943 SetToFirstInsertPoint(if_data.true_block, b_); in EmitErfInv()
955 SetToFirstInsertPoint(if_data.false_block, b_); in EmitErfInv()
969 SetToFirstInsertPoint(if_data.after_block, b_); in EmitErfInv()
973 llvm_ir::LlvmIfData if_data = llvm_ir::EmitIfThenElse( in EmitErfInv() local
976 SetToFirstInsertPoint(if_data.true_block, b_); in EmitErfInv()
996 SetToFirstInsertPoint(if_data.false_block, b_); in EmitErfInv()
1040 SetToFirstInsertPoint(if_data.after_block, b_); in EmitErfInv()
2024 llvm_ir::LlvmIfData if_data = in EmitElementalDynamicUpdateSlice() local
2028 SetToFirstInsertPoint(if_data.true_block, b_); in EmitElementalDynamicUpdateSlice()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc423 llvm_ir::LlvmIfData if_data = in MakeElementGenerator() local
425 SetToFirstInsertPoint(if_data.true_block, b_); in MakeElementGenerator()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc710 llvm_ir::LlvmIfData if_data = in EmitElementalReduceWindow() local
712 SetToFirstInsertPoint(if_data.true_block, &b_); in EmitElementalReduceWindow()
1052 llvm_ir::LlvmIfData if_data = in EmitElementalConvolution() local
1054 SetToFirstInsertPoint(if_data.true_block, &b_); in EmitElementalConvolution()
2565 llvm_ir::LlvmIfData if_data = in HandleConditional() local
2568 SetToFirstInsertPoint(if_data.true_block, &b_); in HandleConditional()
2572 SetToFirstInsertPoint(if_data.false_block, &b_); in HandleConditional()
2576 SetToFirstInsertPoint(if_data.after_block, &b_); in HandleConditional()