Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc90 using llvm_ir::SetToFirstInsertPoint;
736 SetToFirstInsertPoint(source_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
760 SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), &b_); in HandleSelectAndScatter()
784 SetToFirstInsertPoint(if_in_bounds.true_block, &b_); in HandleSelectAndScatter()
792 SetToFirstInsertPoint(if_initialized.false_block, &b_); in HandleSelectAndScatter()
813 SetToFirstInsertPoint(if_initialized.true_block, &b_); in HandleSelectAndScatter()
832 SetToFirstInsertPoint(if_select_lhs.false_block, &b_); in HandleSelectAndScatter()
841 SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
865 SetToFirstInsertPoint(source_loops.GetOuterLoopExitBasicBlock(), &b_); in HandleSelectAndScatter()
1662 SetToFirstInsertPoint(reduction_loop_nest.GetInnerLoopBodyBasicBlock(), &b_); in EmitInnerLoopForVectorizedReduction()
[all …]
H A Dparallel_loop_emitter.cc74 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_); in EmitIndexAndSetExitBasicBlock()
H A Ddot_op_emitter.cc56 using llvm_ir::SetToFirstInsertPoint;
678 SetToFirstInsertPoint(&func->getEntryBlock(), b_); in EmitNaiveLlvmIrGemm()
695 SetToFirstInsertPoint(reduction_loop->GetBodyBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
730 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Delemental_ir_emitter.cc56 using llvm_ir::SetToFirstInsertPoint;
1931 llvm_ir::SetToFirstInsertPoint(exit_block, b_); in EmitElementalConcatenate()
1970 llvm_ir::SetToFirstInsertPoint(emit_operand_blocks[operand_id], b_); in EmitElementalConcatenate()
2280 SetToFirstInsertPoint(if_data.true_block, b_); in EmitElementalDynamicUpdateSlice()
2293 SetToFirstInsertPoint(if_data.false_block, b_); in EmitElementalDynamicUpdateSlice()
2298 SetToFirstInsertPoint(if_data.after_block, b_); in EmitElementalDynamicUpdateSlice()
2342 SetToFirstInsertPoint(if_data.true_block, b_); in EmitElementalPad()
2349 SetToFirstInsertPoint(if_data.false_block, b_); in EmitElementalPad()
2355 SetToFirstInsertPoint(if_data.after_block, b_); in EmitElementalPad()
2388 SetToFirstInsertPoint(inner_loop->GetPreheaderBasicBlock(), b_); in EmitElementalDot()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dparallel_loop_emitter.cc197 llvm_ir::SetToFirstInsertPoint(body_block, b_); in EmitLogicalIndexAndSetExitBasicBlock()
293 llvm_ir::SetToFirstInsertPoint(if_in_bounds.true_block, b_); in EmitIndexAndSetExitBasicBlock()
H A Dir_emitter.cc68 using llvm_ir::SetToFirstInsertPoint;
445 SetToFirstInsertPoint(loop_exit_bb, &b_); in EmitAtomicOperationUsingCAS()
H A Dir_emitter_unnested.cc832 llvm_ir::SetToFirstInsertPoint(if_in_dyn_bounds.true_block, &b_); in EmitPadToStatic()
954 llvm_ir::SetToFirstInsertPoint(if_in_dyn_bounds.true_block, &b_); in EmitSliceToDynamic()
2010 llvm_ir::SetToFirstInsertPoint(window_loops.GetInnerLoopBodyBasicBlock(), in EmitSelectAndScatter()
2042 llvm_ir::SetToFirstInsertPoint(if_in_bounds.true_block, &b_); in EmitSelectAndScatter()
2050 llvm_ir::SetToFirstInsertPoint(if_initialized.false_block, &b_); in EmitSelectAndScatter()
2070 llvm_ir::SetToFirstInsertPoint(if_initialized.true_block, &b_); in EmitSelectAndScatter()
2097 llvm_ir::SetToFirstInsertPoint(if_select_lhs.false_block, &b_); in EmitSelectAndScatter()
2105 llvm_ir::SetToFirstInsertPoint(window_loops.GetOuterLoopExitBasicBlock(), in EmitSelectAndScatter()
2111 llvm_ir::SetToFirstInsertPoint(if_should_store.true_block, &b_); in EmitSelectAndScatter()
2407 llvm_ir::SetToFirstInsertPoint(if_window_in_bounds_data.true_block, &b_); in EmitScatter()
H A Delemental_ir_emitter.cc57 using llvm_ir::SetToFirstInsertPoint;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dllvm_util.h241 void SetToFirstInsertPoint(llvm::BasicBlock* blk, llvm::IRBuilder<>* builder);
H A Dllvm_util.cc480 void SetToFirstInsertPoint(llvm::BasicBlock* blk, llvm::IRBuilder<>* builder) { in SetToFirstInsertPoint() function