Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc56 using llvm_ir::SetToFirstInsertPoint;
386 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b_); in MakeElementGenerator()
425 SetToFirstInsertPoint(if_data.true_block, b_); in MakeElementGenerator()
437 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b_); in MakeElementGenerator()
473 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), b()); in MakeElementGenerator()
482 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), b()); in MakeElementGenerator()
Dir_emitter.cc53 using llvm_ir::SetToFirstInsertPoint;
394 SetToFirstInsertPoint(loop_exit_bb, &b_); in EmitAtomicOperationUsingCAS()
622 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), &b_); in HandleDot()
693 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleReduce()
717 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in HandleReduce()
Dparallel_loop_emitter.cc143 llvm_ir::SetToFirstInsertPoint(if_in_bounds.true_block, b_); in EmitIndexAndSetExitBasicBlock()
Dir_emitter_unnested.cc860 llvm_ir::SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), in HandleSelectAndScatter()
885 llvm_ir::SetToFirstInsertPoint(if_in_bounds.true_block, &b_); in HandleSelectAndScatter()
891 llvm_ir::SetToFirstInsertPoint(if_initialized.false_block, &b_); in HandleSelectAndScatter()
911 llvm_ir::SetToFirstInsertPoint(if_initialized.true_block, &b_); in HandleSelectAndScatter()
933 llvm_ir::SetToFirstInsertPoint(if_select_lhs.false_block, &b_); in HandleSelectAndScatter()
941 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in HandleSelectAndScatter()
1172 llvm_ir::SetToFirstInsertPoint(if_window_in_bounds_data.true_block, &b_); in EmitScatter()
2797 llvm_ir::SetToFirstInsertPoint(if_lane_id_is_zero_data.true_block, &b_); in EmitEpilogueForReduction()
2806 llvm_ir::SetToFirstInsertPoint(if_output_inbound_data.true_block, &b_); in EmitEpilogueForReduction()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc78 using llvm_ir::SetToFirstInsertPoint;
666 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), &b_); in EmitElementalReduceWindow()
712 SetToFirstInsertPoint(if_data.true_block, &b_); in EmitElementalReduceWindow()
724 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), &b_); in EmitElementalReduceWindow()
794 SetToFirstInsertPoint(source_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
817 SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
841 SetToFirstInsertPoint(if_in_bounds.true_block, &b_); in HandleSelectAndScatter()
847 SetToFirstInsertPoint(if_initialized.false_block, &b_); in HandleSelectAndScatter()
867 SetToFirstInsertPoint(if_initialized.true_block, &b_); in HandleSelectAndScatter()
883 SetToFirstInsertPoint(if_select_lhs.false_block, &b_); in HandleSelectAndScatter()
[all …]
Dparallel_loop_emitter.cc68 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_); in EmitIndexAndSetExitBasicBlock()
Ddot_op_emitter.cc44 using llvm_ir::SetToFirstInsertPoint;
471 SetToFirstInsertPoint(&func->getEntryBlock(), b_); in EmitNaiveLlvmIrGemm()
488 SetToFirstInsertPoint(reduction_loop->GetBodyBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
517 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc54 using llvm_ir::SetToFirstInsertPoint;
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()
976 SetToFirstInsertPoint(if_data.true_block, b_); in EmitErfInv()
996 SetToFirstInsertPoint(if_data.false_block, b_); in EmitErfInv()
999 SetToFirstInsertPoint(if_data_second.true_block, b_); in EmitErfInv()
1020 SetToFirstInsertPoint(if_data_second.false_block, b_); in EmitErfInv()
1040 SetToFirstInsertPoint(if_data.after_block, b_); in EmitErfInv()
1729 llvm_ir::SetToFirstInsertPoint(exit_block, b_); in EmitElementalConcatenate()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.h253 void SetToFirstInsertPoint(llvm::BasicBlock* blk, llvm::IRBuilder<>* builder);
Dllvm_util.cc476 void SetToFirstInsertPoint(llvm::BasicBlock* blk, llvm::IRBuilder<>* builder) { in SetToFirstInsertPoint() function