Home
last modified time | relevance | path

Searched refs:SetToFirstInsertPoint (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc86 using llvm_ir::SetToFirstInsertPoint;
713 SetToFirstInsertPoint(source_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
736 SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
760 SetToFirstInsertPoint(if_in_bounds.true_block, &b_); in HandleSelectAndScatter()
766 SetToFirstInsertPoint(if_initialized.false_block, &b_); in HandleSelectAndScatter()
786 SetToFirstInsertPoint(if_initialized.true_block, &b_); in HandleSelectAndScatter()
802 SetToFirstInsertPoint(if_select_lhs.false_block, &b_); in HandleSelectAndScatter()
810 SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
830 SetToFirstInsertPoint(source_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
1564 SetToFirstInsertPoint(reduction_loop_nest.GetInnerLoopBodyBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()
[all …]
Dparallel_loop_emitter.cc74 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_); in EmitIndexAndSetExitBasicBlock()
Ddot_op_emitter.cc56 using llvm_ir::SetToFirstInsertPoint;
614 SetToFirstInsertPoint(&func->getEntryBlock(), b_); in EmitNaiveLlvmIrGemm()
631 SetToFirstInsertPoint(reduction_loop->GetBodyBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
666 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc56 using llvm_ir::SetToFirstInsertPoint;
1776 llvm_ir::SetToFirstInsertPoint(exit_block, b_); in EmitElementalConcatenate()
1815 llvm_ir::SetToFirstInsertPoint(emit_operand_blocks[operand_id], b_); in EmitElementalConcatenate()
2087 SetToFirstInsertPoint(if_data.true_block, b_); in EmitElementalDynamicUpdateSlice()
2100 SetToFirstInsertPoint(if_data.false_block, b_); in EmitElementalDynamicUpdateSlice()
2105 SetToFirstInsertPoint(if_data.after_block, b_); in EmitElementalDynamicUpdateSlice()
2149 SetToFirstInsertPoint(if_data.true_block, b_); in EmitElementalPad()
2156 SetToFirstInsertPoint(if_data.false_block, b_); in EmitElementalPad()
2162 SetToFirstInsertPoint(if_data.after_block, b_); in EmitElementalPad()
2195 SetToFirstInsertPoint(inner_loop->GetPreheaderBasicBlock(), b_); in EmitElementalDot()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc153 llvm_ir::SetToFirstInsertPoint(if_in_bounds.true_block, b_); in EmitIndexAndSetExitBasicBlock()
Dir_emitter.cc75 using llvm_ir::SetToFirstInsertPoint;
493 SetToFirstInsertPoint(loop_exit_bb, &b_); in EmitAtomicOperationUsingCAS()
Delemental_ir_emitter.cc58 using llvm_ir::SetToFirstInsertPoint;
Dir_emitter_unnested.cc914 llvm_ir::SetToFirstInsertPoint(if_in_dyn_bounds.true_block, &b_); in EmitPadToStaticFromMlir()
1035 llvm_ir::SetToFirstInsertPoint(if_in_dyn_bounds.true_block, &b_); in EmitSliceToDynamicFromMlir()
2298 llvm_ir::SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), in EmitSelectAndScatterFromMlir()
2330 llvm_ir::SetToFirstInsertPoint(if_in_bounds.true_block, &b_); in EmitSelectAndScatterFromMlir()
2336 llvm_ir::SetToFirstInsertPoint(if_initialized.false_block, &b_); in EmitSelectAndScatterFromMlir()
2355 llvm_ir::SetToFirstInsertPoint(if_initialized.true_block, &b_); in EmitSelectAndScatterFromMlir()
2383 llvm_ir::SetToFirstInsertPoint(if_select_lhs.false_block, &b_); in EmitSelectAndScatterFromMlir()
2391 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in EmitSelectAndScatterFromMlir()
2694 llvm_ir::SetToFirstInsertPoint(if_window_in_bounds_data.true_block, &b_); in EmitScatter()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.h248 void SetToFirstInsertPoint(llvm::BasicBlock* blk, llvm::IRBuilder<>* builder);
Dllvm_util.cc479 void SetToFirstInsertPoint(llvm::BasicBlock* blk, llvm::IRBuilder<>* builder) { in SetToFirstInsertPoint() function